Searched defs:root_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 42 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
80 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); variable
111 auto root_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
138 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
186 auto root_shape = ShapeUtil::MakeShape(F32, {2, 3}); variable
231 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); variable
268 auto root_shape = ShapeUtil::MakeShape(F32, {3, 2}); variable
321 auto root_shape = ShapeUtil::MakeShape(F32, {2, 3}); variable
361 auto root_shape = ShapeUtil::MakeShape(F32, {1, 1, 3}); variable
390 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 65 const Shape& root_shape = root->shape(); in EmitBasePointersForNestedComputation() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1413 const Shape& root_shape = root_instruction->shape(); in MatchReductionGenerator() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 296 TensorShapeProto root_shape; member
|