Searched refs:constv (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 206 auto constv = builder.AddInstruction( in BuildWhileBodyComputation() local
215 HloInstruction::CreateGetTupleElement(constv->shape(), param, 1)); in BuildWhileBodyComputation()
217 constv->shape(), HloOpcode::kAdd, indexv, constv)); in BuildWhileBodyComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 222 auto constv = builder.AddInstruction(HloInstruction::CreateConstant( in BuildWhileBodyComputation() local
231 HloInstruction::CreateGetTupleElement(constv->shape(), param, 1)); in BuildWhileBodyComputation()
233 constv->shape(), HloOpcode::kAdd, indexv, constv)); in BuildWhileBodyComputation()
|
| D | memory_space_assignment_test.cc | 3724 auto constv = builder.AddInstruction( in TEST_P() local
3733 HloInstruction::CreateGetTupleElement(constv->shape(), param, 1)); in TEST_P()
3735 constv->shape(), HloOpcode::kAdd, indexv, constv)); in TEST_P()
3739 schedule.set_sequence(body_computation, {const1, constv, param, indexc, in TEST_P()
|