Searched refs:induction_variable_shape_ (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | while_transformer_test.cc | 33 induction_variable_shape_(ShapeUtil::MakeShape(S32, {})), in WhileTransformerTest()
63 induction_variable_shape_, loop_state, ind_var_tuple_index)); in BuildBodyComputation()
112 {induction_variable_shape_, data_shape_}); in GetLoopStateShape()
115 {data_shape_, induction_variable_shape_}); in GetLoopStateShape()
120 Shape induction_variable_shape_; member in xla::__anon7d016b820111::WhileTransformerTest
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | copy_insertion_test.cc | 373 induction_variable_shape_, loop_state, 0)); in BuildDependentBodyComputation()
406 {induction_variable_shape_, data_shape_, data_shape_}); in BuildDependentBodyComputation2()
414 induction_variable_shape_, loop_state, 0)); in BuildDependentBodyComputation2()
449 induction_variable_shape_, loop_state, 0)); in BuildDependentBodyOneReadOnlyComputation()
486 induction_variable_shape_, loop_state, 0)); in BuildIndependentBodyComputation()
536 induction_variable_shape_, loop_state, 0)); in BuildNestedBodyComputation()
684 Shape induction_variable_shape_ = ShapeUtil::MakeShape(S32, {}); member in xla::__anon2db0cb110111::WhileCopyInsertionTest
687 ShapeUtil::MakeTupleShape({induction_variable_shape_, data_shape_});
691 {induction_variable_shape_, nested_tuple_shape_});
1001 HloInstruction::CreateParameter(0, induction_variable_shape_, "iter")); in TEST_F()
[all …]
|