| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | reshape_mover_test.cc | 43 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
81 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); variable
112 auto root_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
139 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
187 auto root_shape = ShapeUtil::MakeShape(F32, {2, 3}); variable
232 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); variable
269 auto root_shape = ShapeUtil::MakeShape(F32, {3, 2}); variable
322 auto root_shape = ShapeUtil::MakeShape(F32, {2, 3}); variable
362 auto root_shape = ShapeUtil::MakeShape(F32, {1, 1, 3}); variable
391 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
[all …]
|
| H A D | conditional_canonicalizer.cc | 35 const Shape& root_shape = conditional->shape(); in CanonicalizeNonTupleConditional() local
|
| H A D | hlo_verifier.cc | 1490 const Shape& root_shape = async_done->operand(0)->shape().tuple_shapes(1); in HandleAsyncDone() local
|
| H A D | algebraic_simplifier_test.cc | 7227 const Shape root_shape = m->entry_computation()->root_instruction()->shape(); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | dynamic_shaped_ops.cc | 140 const XlaComputation& computation) -> StatusOr<XlaComputation> { in DynamicConditional()
200 const XlaComputation& computation) -> StatusOr<XlaComputation> { in DynamicConditional()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_nested.cc | 82 const Shape& root_shape = root->shape(); in CodegenNestedComputation() local
|
| H A D | ir_emitter_unnested.cc | 4843 const Shape& root_shape = fused_computation->root_instruction()->shape(); in GroupDisjointReductions() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder_test.cc | 635 const Shape& root_shape = in TEST_F() local
648 const Shape& root_shape = in TEST_F() local
664 const Shape& root_shape = in TEST_F() local
|
| H A D | value_inference.cc | 1302 Shape root_shape = Shape(root->shape()); in AnalyzeIsDynamic() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_executable.cc | 229 const Shape& root_shape = root->shape(); in CreateResultShapedBuffer() local
|
| H A D | ir_emitter.cc | 1472 const Shape& root_shape = root_instruction->shape(); in MatchReductionGenerator() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | hlo_control_flow_flattening_test.cc | 278 Shape root_shape = module->entry_computation()->root_instruction()->shape(); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer.cc | 383 TensorShapeProto root_shape; member
|