Home
last modified time | relevance | path

Searched refs:data_shape_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc34 data_shape_(ShapeUtil::MakeShape(F32, {8})),
70 data_shape_, loop_state, data_tuple_index)); in BuildBodyComputation()
75 HloInstruction::CreateBroadcast(data_shape_, cast, {})); in BuildBodyComputation()
77 data_shape_, HloOpcode::kAdd, data, update)); in BuildBodyComputation()
112 {induction_variable_shape_, data_shape_}); in GetLoopStateShape()
115 {data_shape_, induction_variable_shape_}); in GetLoopStateShape()
121 Shape data_shape_; member in xla::__anon9d1b4d8c0111::WhileTransformerTest
/external/tensorflow/tensorflow/compiler/xla/service/
Dcopy_insertion_test.cc451 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation()
457 HloInstruction::CreateBroadcast(data_shape_, convert, {})); in BuildDependentBodyComputation()
459 data_shape_, HloOpcode::kAdd, data, update)); in BuildDependentBodyComputation()
477 {induction_variable_shape_, data_shape_, data_shape_}); in BuildDependentBodyComputation2()
494 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation2()
498 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 2)); in BuildDependentBodyComputation2()
523 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyOneReadOnlyComputation()
530 HloInstruction::CreateBroadcast(data_shape_, convert, {})); in BuildDependentBodyOneReadOnlyComputation()
532 data_shape_, HloOpcode::kAdd, data, update)); in BuildDependentBodyOneReadOnlyComputation()
569 HloInstruction::CreateGetTupleElement(data_shape_, data, 0)); in BuildIndependentBodyComputation()
[all …]
Dbuffer_assignment_test.cc2027 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 0)); in BuildWhileBodyComputation()
2029 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildWhileBodyComputation()
2031 data_shape_, HloOpcode::kMultiply, input, weights)); in BuildWhileBodyComputation()
2053 Shape data_shape_ = ShapeUtil::MakeShape(F32, {4}); member in xla::__anon3198ac380111::WhileBufferAssignmentTest
2055 ShapeUtil::MakeTupleShape({data_shape_, data_shape_, data_shape_});
2068 HloInstruction::CreateParameter(0, data_shape_, "input0")); in TEST_F()
2070 HloInstruction::CreateParameter(1, data_shape_, "weights0")); in TEST_F()
2072 HloInstruction::CreateParameter(2, data_shape_, "weights1")); in TEST_F()
2077 HloInstruction::CreateBroadcast(data_shape_, zero, {})); in TEST_F()
2079 HloInstruction::CreateBroadcast(data_shape_, zero, {})); in TEST_F()
[all …]