Searched refs:PositionInContainer (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 379 EXPECT_GT(PositionInContainer(log_minor_to_major, 1), in TEST_F()
380 PositionInContainer(log_minor_to_major, 2)); in TEST_F()
384 EXPECT_GT(PositionInContainer(reshape_minor_to_major, 0), in TEST_F()
385 PositionInContainer(reshape_minor_to_major, 2)); in TEST_F()
|
| D | algebraic_simplifier.cc | 5460 (PositionInContainer(LayoutUtil::MinorToMajor(filter_shape), in SimplifyConvToDot()
5462 PositionInContainer(LayoutUtil::MinorToMajor(filter_shape), in SimplifyConvToDot()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 431 PositionInContainer(layout.minor_to_major(), dim)); in AreDimensionsConsecutive()
|
| D | util.h | 332 int64 PositionInContainer(const Container& container, int64 value) { in PositionInContainer() function
|
| D | shape_util.cc | 1701 permutation[i] = PositionInContainer(major_to_minor_b, major_to_minor_a[i]); in FindTranspose021()
|