Searched defs:flat_shape (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | all_reduce_contiguous.cc | 45 Shape flat_shape = ShapeUtil::MakeShape(element_type, {num_elements}); in ReplaceWithContiguousAllReduce() local
68 const Shape& flat_shape = flat_operands[i]->shape(); in ReplaceWithContiguousAllReduce() local
|
| H A D | while_loop_simplifier.cc | 1039 const Shape& flat_shape = instr->shape(); in TryFlattenNestedTuples() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | rsqrt_builder.cc | 55 int flat_shape[] = {1, 1, 1, num_elements_}; in AddNumerator() local
91 int flat_shape[] = {1, 1, 1, num_elements_}; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| H A D | restore_ops.cc | 52 int64_t flat_shape[] = {1}; in CreateStringVectorTensorHandle() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| H A D | scatter_nd_fuzz.cc | 66 auto flat_shape = shape_tensor.flat<int32>(); in FuzzImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | analytical_cost_estimator_test.cc | 71 auto flat_shape = ops::Const(s.WithOpName("flat_shape"), in CreateMiniGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | all_reduce_blueconnect.cc | 215 Shape flat_shape = ShapeUtil::MakeShape(element_type, {num_elements}); in TryDecomposeAllReduce() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | decode_proto_op.cc | 792 TensorShape flat_shape = {static_cast<int64_t>(message_count), in Compute() local
|