Searched refs:shape4 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_test.cc | 2526 Shape shape4 = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_P() local
2528 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape3, shape4}); in TEST_P()
2532 HloInstruction::CreateGetTupleElement(shape4, p0, 1)); in TEST_P()
2564 HloInstruction::CreateUnary(shape4, HloOpcode::kNegate, p4)); in TEST_P()
2568 HloInstruction::CreatePad(shape4, add3, padding_value, padding_config)); in TEST_P()
2570 shape4, HloOpcode::kAdd, padded_add3, negate7)); in TEST_P()
|