Searched refs:induction_variable (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_transformer_test.cc | 44 auto induction_variable = in BuildConditionComputation() local
48 condition_result_shape_, induction_variable, limit_const, in BuildConditionComputation()
61 auto induction_variable = in BuildBodyComputation() local
67 induction_variable->shape(), HloOpcode::kAdd, induction_variable, inc)); in BuildBodyComputation()
73 ShapeUtil::MakeShape(F32, {}), induction_variable)); in BuildBodyComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_control_flow_flattening.cc | 195 HloInstruction* induction_variable = in FlattenWhileLoop() local
200 ShapeUtil::MakeShape(PRED, {}), induction_variable, limit, in FlattenWhileLoop()
214 HloInstruction* induction_variable = in FlattenWhileLoop() local
219 induction_variable = body->AddInstruction(HloInstruction::CreateBinary( in FlattenWhileLoop()
220 shape, HloOpcode::kAdd, induction_variable, increment)); in FlattenWhileLoop()
222 TupleUtil::AppendSuffix(old_root, {induction_variable}); in FlattenWhileLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 349 auto induction_variable = in BuildConditionComputation() local
353 condition_result_shape_, induction_variable, limit_const, in BuildConditionComputation()
371 auto induction_variable = in BuildDependentBodyComputation() local
377 induction_variable->shape(), HloOpcode::kAdd, induction_variable, inc)); in BuildDependentBodyComputation()
384 HloInstruction::CreateConvert(f32_scalar_shape, induction_variable)); in BuildDependentBodyComputation()
412 auto induction_variable = in BuildDependentBodyComputation2() local
420 induction_variable->shape(), HloOpcode::kAdd, induction_variable, inc)); in BuildDependentBodyComputation2()
447 auto induction_variable = in BuildDependentBodyOneReadOnlyComputation() local
457 HloInstruction::CreateConvert(f32_scalar_shape, induction_variable)); in BuildDependentBodyOneReadOnlyComputation()
464 HloInstruction::CreateTuple({induction_variable, add1})); in BuildDependentBodyOneReadOnlyComputation()
[all …]
|