HomeSort by relevance Sort by last modified time
    Searched refs:induction_variable_shape_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
while_transformer_test.cc 33 induction_variable_shape_(ShapeUtil::MakeShape(S32, {})),
63 induction_variable_shape_, loop_state, ind_var_tuple_index));
112 {induction_variable_shape_, data_shape_});
115 {data_shape_, induction_variable_shape_});
120 Shape induction_variable_shape_; member in class:xla::__anon44447::WhileTransformerTest
  /external/tensorflow/tensorflow/compiler/xla/service/
copy_insertion_test.cc 444 induction_variable_shape_, loop_state, 0));
474 {induction_variable_shape_, data_shape_, data_shape_});
482 induction_variable_shape_, loop_state, 0));
517 induction_variable_shape_, loop_state, 0));
551 induction_variable_shape_, loop_state, 0));
601 induction_variable_shape_, loop_state, 0));
773 Shape induction_variable_shape_ = ShapeUtil::MakeShape(S32, {}); member in class:xla::__anon44322::WhileCopyInsertionTest
    [all...]

Completed in 53 milliseconds