Searched refs:induction_variable (Results 1 – 2 of 2) 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/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()
454 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8})); in BuildDependentBodyComputation()
480 auto induction_variable = in BuildDependentBodyComputation2() local
488 induction_variable->shape(), HloOpcode::kAdd, induction_variable, inc)); in BuildDependentBodyComputation2()
515 auto induction_variable = in BuildDependentBodyOneReadOnlyComputation() local
524 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8})); in BuildDependentBodyOneReadOnlyComputation()
529 HloInstruction::CreateTuple({induction_variable, add1})); in BuildDependentBodyOneReadOnlyComputation()
[all …]
|