| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| while_transformer_test.cc | 34 data_shape_(ShapeUtil::MakeShape(F32, {8})),
70 data_shape_, loop_state, data_tuple_index));
75 HloInstruction::CreateBroadcast(data_shape_, cast, {}));
77 data_shape_, HloOpcode::kAdd, data, update));
112 {induction_variable_shape_, data_shape_});
115 {data_shape_, induction_variable_shape_});
121 Shape data_shape_; member in class:xla::__anon44447::WhileTransformerTest
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| copy_insertion_test.cc | 451 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1));
454 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8}));
456 data_shape_, HloOpcode::kAdd, data, update));
474 {induction_variable_shape_, data_shape_, data_shape_});
491 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1));
495 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 2));
520 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1));
524 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8}));
526 data_shape_, HloOpcode::kAdd, data, update))
774 Shape data_shape_ = ShapeUtil::MakeShape(F32, {8}); member in class:xla::__anon44322::WhileCopyInsertionTest
[all...] |
| buffer_assignment_test.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| range_coder_ops.cc | 54 std::copy(data_shape.begin(), data_shape.end(), &data_shape_[0]);
91 if (data_index_[i] < data_shape_[i]) {
110 std::array<int64, N> data_shape_; member in class:tensorflow::__anon44736::BroadcastRange
|
