Searched refs:induction_variable_shape_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| 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::__anon9d1b4d8c0111::WhileTransformerTest
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 444 induction_variable_shape_, loop_state, 0)); in BuildDependentBodyComputation()
474 {induction_variable_shape_, data_shape_, data_shape_}); in BuildDependentBodyComputation2()
482 induction_variable_shape_, loop_state, 0)); in BuildDependentBodyComputation2()
517 induction_variable_shape_, loop_state, 0)); in BuildDependentBodyOneReadOnlyComputation()
551 induction_variable_shape_, loop_state, 0)); in BuildIndependentBodyComputation()
601 induction_variable_shape_, loop_state, 0)); in BuildNestedBodyComputation()
773 Shape induction_variable_shape_ = ShapeUtil::MakeShape(S32, {}); member in xla::__anon1fa5445b0111::WhileCopyInsertionTest
776 ShapeUtil::MakeTupleShape({induction_variable_shape_, data_shape_});
780 {induction_variable_shape_, nested_tuple_shape_});
894 HloInstruction::CreateParameter(0, induction_variable_shape_, "iter")); in TEST_F()
[all …]
|