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 | 98 HloInstruction* induction_variable = in FlattenWhileLoop() local
103 ShapeUtil::MakeShape(PRED, {}), induction_variable, limit, in FlattenWhileLoop()
115 HloInstruction* induction_variable = in FlattenWhileLoop() local
120 induction_variable = body->AddInstruction(HloInstruction::CreateBinary( in FlattenWhileLoop()
121 shape, HloOpcode::kAdd, induction_variable, increment)); in FlattenWhileLoop()
123 TupleUtil::AppendSuffix(old_root, {induction_variable}); in FlattenWhileLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 420 auto induction_variable = in BuildConditionComputation() local
424 condition_result_shape_, induction_variable, limit_const, in BuildConditionComputation()
442 auto induction_variable = in BuildDependentBodyComputation() local
448 induction_variable->shape(), HloOpcode::kAdd, induction_variable, inc)); in BuildDependentBodyComputation()
455 HloInstruction::CreateConvert(f32_scalar_shape, induction_variable)); in BuildDependentBodyComputation()
483 auto induction_variable = in BuildDependentBodyComputation2() local
491 induction_variable->shape(), HloOpcode::kAdd, induction_variable, inc)); in BuildDependentBodyComputation2()
518 auto induction_variable = in BuildDependentBodyOneReadOnlyComputation() local
528 HloInstruction::CreateConvert(f32_scalar_shape, induction_variable)); in BuildDependentBodyOneReadOnlyComputation()
535 HloInstruction::CreateTuple({induction_variable, add1})); in BuildDependentBodyOneReadOnlyComputation()
[all …]
|