HomeSort by relevance Sort by last modified time
    Searched refs:minor_to_major_size (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
layout.cc 51 layout.minor_to_major_.reserve(proto.minor_to_major_size());
66 proto.mutable_minor_to_major()->Reserve(minor_to_major_size());
layout_util.cc 212 if (layout.minor_to_major_size() != 0) {
226 if (layout.minor_to_major_size() != shape.rank()) {
230 layout.minor_to_major_size(), shape.rank(),
346 CHECK_LT(physical_dimension_number, layout.minor_to_major_size());
348 layout.minor_to_major_size() - 1 - physical_dimension_number);
355 CHECK_LT(physical_dimension_number, layout.minor_to_major_size());
361 std::vector<int64> logical_to_physical(layout.minor_to_major_size());
layout.h 153 int minor_to_major_size() const { return minor_to_major_.size(); } function in class:xla::Layout
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
shape_partition.cc 29 for (int i = shape_.layout().minor_to_major_size() - 1; i >= 0; --i) {
109 shape_.layout().minor_to_major_size() - 1 - i);
shape_partition_test.cc 174 shape.layout().minor_to_major_size() - 1 - i);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cusolver_rewriter.cc 44 int n = shape->mutable_layout()->minor_to_major_size();
gpu_layout_assignment.cc 252 int n = shape->mutable_layout()->minor_to_major_size();
ir_emitter_unnested.cc 542 int n = layout.minor_to_major_size();
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.cc 45 for (int64 i = 0; i < layout.minor_to_major_size(); ++i) {
60 if (i < layout.minor_to_major_size() - 1) {
  /external/tensorflow/tensorflow/compiler/xla/tests/
copy_test.cc 135 ASSERT_EQ(2, literal_layout->minor_to_major_size());
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_parser.cc     [all...]

Completed in 716 milliseconds