| /external/tensorflow/tensorflow/compiler/xla/ |
| layout_util_test.cc | 261 Shape shape_with_layout = ShapeUtil::MakeShape(F32, {2, 3, 4}); local
262 LayoutUtil::SetToDefaultLayout(&shape_with_layout);
263 EXPECT_FALSE(LayoutUtil::IsPadded(shape_with_layout));
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_layout_assignment.cc | 213 Shape shape_with_layout = buf.shape(); local
214 *shape_with_layout.mutable_layout() = buffer_constraint.layout();
223 shape_with_layout, instruction, /*operand_no=*/0));
231 shape_with_layout, instruction, /*operand_no=*/0));
239 shape_with_layout, instruction, /*operand_no=*/0));
241 shape_with_layout, instruction, /*operand_no=*/4));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_cost_analysis_test.cc | 397 Shape shape_with_layout = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); local
399 Shape shape_without_layout = shape_with_layout;
411 shape_with_layout, HloOpcode::kAdd, c1, broadcast));
|
| layout_assignment.cc | 64 const Shape& shape_with_layout, HloInstruction* instruction) {
65 TF_RET_CHECK(LayoutUtil::HasLayout(shape_with_layout));
66 DCHECK(ShapeUtil::Compatible(shape_with_layout, instruction->shape()))
67 << ShapeUtil::HumanString(shape_with_layout) << " "
85 ShapeUtil::GetSubshape(shape_with_layout, {i}), gte));
93 shape_with_layout, tuple_copy->mutable_shape()));
101 shape_with_layout, copy->mutable_shape()));
267 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout,
273 << ShapeUtil::HumanStringWithLayout(shape_with_layout);
279 shape_with_layout)) {
465 Shape const* shape_with_layout = nullptr; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| literal_test_util.cc | 772 Shape shape_with_layout = new_literal->shape(); local
773 *shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major);
780 IndexUtil::LinearIndexToMultidimensionalIndex(shape_with_layout, i);
|
| local_client_execute_test.cc | 273 Shape shape_with_layout = ShapeUtil::MakeTupleShape( local
278 options.set_result_layout(shape_with_layout);
[all...] |
