| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_layout.cc | 66 *shape_.mutable_layout() = layout;
73 *ShapeUtil::GetMutableSubshape(&shape_, shape_index)->mutable_layout() =
|
| layout_util_test.cc | 33 *shape.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major);
41 *shape.mutable_layout() = LayoutUtil::MakeSparseLayout(max_sparse_elements);
211 *src.mutable_layout() = LayoutUtil::MakeLayout({1, 2, 3, 4});
328 tile = shape.mutable_layout()->add_tiles();
335 shape.mutable_layout()->clear_tiles();
336 tile = shape.mutable_layout()->add_tiles();
343 tile = shape.mutable_layout()->add_tiles();
346 tile = shape.mutable_layout()->add_tiles();
354 tile = shape.mutable_layout()->add_tiles();
361 shape.mutable_layout()->clear_tiles()
[all...] |
| packed_literal_reader.cc | 50 *literal_shape.mutable_layout() = *layout;
|
| index_util_test.cc | 28 shape->mutable_layout()->clear_minor_to_major();
30 shape->mutable_layout()->add_minor_to_major(dimension);
|
| shape.cc | 54 *mutable_layout() = Layout::CreateFromProto(shape_proto.layout());
73 *proto.mutable_layout() = layout().ToProto();
|
| layout_util.cc | 150 shape->mutable_layout()->set_format(DENSE);
151 auto* minor_to_major = shape->mutable_layout()->mutable_minor_to_major();
398 *dst->mutable_layout() = src.layout();
|
| shape.h | 126 Layout* mutable_layout() { return &layout_; } function in class:xla::Shape
|
| shape_util.cc | 105 *shape.mutable_layout() =
210 *shape.mutable_layout() = LayoutUtil::MakeSparseLayout(max_sparse_elements);
270 shape->mutable_layout()->add_minor_to_major(shape->rank());
[all...] |
| shape_util_test.cc | 128 auto layout_1 = shape_1.mutable_layout();
134 auto layout_2 = shape_2.mutable_layout();
190 *shape_a.mutable_layout() = Layout({1, 0});
192 *shape_b.mutable_layout() = Layout({0, 1});
194 *shape_c.mutable_layout() = Layout({0, 1});
[all...] |
| literal_util.cc | 309 *shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_hlo_test.cc | 87 *reduce_output_shape->mutable_layout() =
92 *reduce_input_shape->mutable_layout() =
|
| copy_test.cc | 134 Layout* literal_layout = literal.mutable_shape_do_not_use()->mutable_layout();
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_layout_assignment.cc | 137 std::tie(*input_shape->mutable_layout(),
138 *filter_shape->mutable_layout(),
139 *output_shape->mutable_layout()),
225 *shape.mutable_layout() = keys_layout;
252 int n = shape->mutable_layout()->minor_to_major_size();
254 std::swap(shape->mutable_layout()->mutable_minor_to_major()->at(0),
255 shape->mutable_layout()->mutable_minor_to_major()->at(1));
334 *shape_with_layout.mutable_layout() = buffer_constraint.layout();
|
| cusolver_rewriter.cc | 44 int n = shape->mutable_layout()->minor_to_major_size();
46 std::swap(shape->mutable_layout()->mutable_minor_to_major()->at(0),
47 shape->mutable_layout()->mutable_minor_to_major()->at(1));
|
| gpu_layout_assignment_test.cc | 45 *ashape_in_row_major.mutable_layout() = LayoutUtil::MakeLayout({1, 0});
46 *ashape_in_col_major.mutable_layout() = LayoutUtil::MakeLayout({0, 1});
103 *shapes.back().mutable_layout() = LayoutUtil::MakeLayout(layout_vec);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_layout_assignment.cc | 84 *new_shape.mutable_layout() = LayoutUtil::MakeLayout(dimension_order);
92 *new_shape.mutable_layout() = LayoutUtil::MakeLayout(dimension_order);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| layout_assignment_test.cc | 108 *shape.mutable_layout() = layout;
138 *col_major_shape.mutable_layout() = col_major_layout;
143 *row_major_shape.mutable_layout() = row_major_layout;
190 *shape.mutable_layout() = layout;
368 *ashape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({0, 2, 1, 3});
369 *bshape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0});
409 *ashape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({1, 0});
410 *bshape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({0, 1});
446 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({1, 0});
447 *output_shape_with_layout.mutable_layout()
[all...] |
| hlo_cse_test.cc | 433 *exp1->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({0, 1});
437 *exp2->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({1, 0});
464 *exp1->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({0, 1});
468 *exp2->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({1, 0});
|
| layout_assignment.h | 254 *shape.mutable_layout() = it->second;
|
| reshape_mover.cc | 221 *shape->mutable_layout() = new_operand_shape.layout();
|
| layout_assignment.cc | 257 *shape.mutable_layout() = layout;
[all...] |
| algebraic_simplifier_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_cpu.cc | 113 *arg_shape.mutable_layout() =
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| shape_util.cc | 80 *shape->mutable_layout() = layout;
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| ir_array.cc | 124 *b.mutable_layout() = layout_;
|
