| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_transformer_test.cc | 34 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/ |
| D | copy_insertion_test.cc | 451 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation()
454 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8})); in BuildDependentBodyComputation()
456 data_shape_, HloOpcode::kAdd, data, update)); in BuildDependentBodyComputation()
474 {induction_variable_shape_, data_shape_, data_shape_}); in BuildDependentBodyComputation2()
491 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation2()
495 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 2)); in BuildDependentBodyComputation2()
520 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyOneReadOnlyComputation()
524 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8})); in BuildDependentBodyOneReadOnlyComputation()
526 data_shape_, HloOpcode::kAdd, data, update)); in BuildDependentBodyOneReadOnlyComputation()
563 HloInstruction::CreateGetTupleElement(data_shape_, data, 0)); in BuildIndependentBodyComputation()
[all …]
|
| D | buffer_assignment_test.cc | 1878 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 0)); in BuildWhileBodyComputation()
1880 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildWhileBodyComputation()
1882 data_shape_, HloOpcode::kMultiply, input, weights)); in BuildWhileBodyComputation()
1905 Shape data_shape_ = ShapeUtil::MakeShape(F32, {4}); member in xla::__anon3198ac380111::WhileBufferAssignmentTest
1907 ShapeUtil::MakeTupleShape({data_shape_, data_shape_, data_shape_});
1920 HloInstruction::CreateParameter(0, data_shape_, "input0")); in TEST_F()
1922 HloInstruction::CreateParameter(1, data_shape_, "weights0")); in TEST_F()
1924 HloInstruction::CreateParameter(2, data_shape_, "weights1")); in TEST_F()
1929 HloInstruction::CreateBroadcast(data_shape_, zero, {})); in TEST_F()
1931 HloInstruction::CreateBroadcast(data_shape_, zero, {})); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | range_coder_ops.cc | 54 std::copy(data_shape.begin(), data_shape.end(), &data_shape_[0]); in BroadcastRange()
91 if (data_index_[i] < data_shape_[i]) { in Next()
110 std::array<int64, N> data_shape_; member in tensorflow::__anon72f8b5b20111::BroadcastRange
|