Searched refs:elem_shape2 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multioutput_fusion_test.cc | 68 const Shape elem_shape2 = in RunTest2D() local
80 HloInstruction::CreateBroadcast(elem_shape2, add1, {})); in RunTest2D()
83 HloInstruction::CreateParameter(1, elem_shape2, "1")); in RunTest2D()
86 elem_shape2, HloOpcode::kAdd, broadcast, param1)); in RunTest2D()
88 elem_shape2, HloOpcode::kSubtract, param1, broadcast)); in RunTest2D()
93 elem_shape2, sub, add2, dot_dnums, DefaultPrecisionConfig(2))); in RunTest2D()
100 HloInstruction::CreateGetTupleElement(elem_shape2, tuple, 0)); in RunTest2D()
102 HloInstruction::CreateGetTupleElement(elem_shape2, tuple, 1)); in RunTest2D()
|