Home
last modified time | relevance | path

Searched refs:physical_dimension_number (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.cc333 int64 physical_dimension_number) { in Major() argument
334 CHECK_LE(0, physical_dimension_number); in Major()
335 CHECK_LT(physical_dimension_number, layout.minor_to_major_size()); in Major()
337 layout.minor_to_major_size() - 1 - physical_dimension_number); in Major()
341 int64 physical_dimension_number) { in Minor() argument
343 CHECK_LE(0, physical_dimension_number); in Minor()
344 CHECK_LT(physical_dimension_number, layout.minor_to_major_size()); in Minor()
345 return layout.minor_to_major(physical_dimension_number); in Minor()
Dlayout_util.h141 static int64 Major(const Layout& layout, int64 physical_dimension_number);
145 static int64 Minor(const Layout& layout, int64 physical_dimension_number);