OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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