Searched refs:gte_0 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_invariant_code_motion_test.cc | 70 HloInstruction* gte_0 = builder.AddInstruction( in TEST_F() local
76 scalar_s32, HloOpcode::kAdd, gte_0, gte_1)); in TEST_F()
78 HloInstruction::CreateTuple({gte_0, gte_1, add_result})); in TEST_F()
112 HloInstruction* gte_0 = builder.AddInstruction( in TEST_F() local
121 scalar_s32, HloOpcode::kAdd, gte_0, gte_1)); in TEST_F()
137 HloInstruction::CreateTuple({gte_0, gte_1, divide_result})); in TEST_F()
184 HloInstruction* gte_0 = builder.AddInstruction( in TEST_F() local
190 scalar_s32, HloOpcode::kAdd, gte_0, gte_1)); in TEST_F()
191 builder.AddInstruction(HloInstruction::CreateTuple({gte_0, add_result})); in TEST_F()
223 HloInstruction* gte_0 = builder.AddInstruction( in TEST_F() local
[all …]
|
| D | dynamic_dimension_inference_test.cc | 477 auto gte_0 = body_builder.AddInstruction( in TEST_F() local
482 HloInstruction::CreateBinary(input_shape, HloOpcode::kAdd, gte_0, gte_1)); in TEST_F()
|