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

  /external/tensorflow/tensorflow/compiler/xla/
shape.cc 107 layout_.mutable_minor_to_major()->erase(
108 layout_.mutable_minor_to_major()->begin() + i);
112 (*layout_.mutable_minor_to_major())[i] -= 1;
layout.cc 66 proto.mutable_minor_to_major()->Reserve(minor_to_major_size());
layout.h 168 std::vector<int64>* mutable_minor_to_major() { return &minor_to_major_; } function in class:xla::Layout
layout_util.cc 107 std::vector<int64>* minor_to_major = layout.mutable_minor_to_major();
151 auto* minor_to_major = shape->mutable_layout()->mutable_minor_to_major();
shape_util.cc     [all...]
literal_test.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cusolver_rewriter.cc 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.cc 254 std::swap(shape->mutable_layout()->mutable_minor_to_major()->at(0),
255 shape->mutable_layout()->mutable_minor_to_major()->at(1));
  /external/tensorflow/tensorflow/compiler/xla/tests/
copy_test.cc 137 *literal_layout->mutable_minor_to_major() = {

Completed in 98 milliseconds