Home
last modified time | relevance | path

Searched refs:NonSingletonDimensionsAlign (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtranspose_util_test.cc103 EXPECT_TRUE(internal::NonSingletonDimensionsAlign({2, 1, 2}, {1, 0, 2})); in TEST_F()
104 EXPECT_TRUE(internal::NonSingletonDimensionsAlign({2, 1, 2}, {0, 2, 1})); in TEST_F()
105 EXPECT_FALSE(internal::NonSingletonDimensionsAlign({2, 1, 2}, {2, 0, 1})); in TEST_F()
106 EXPECT_FALSE(internal::NonSingletonDimensionsAlign({2, 1, 2}, {2, 1, 0})); in TEST_F()
110 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {0, 2, 4, 1, 3})); in TEST_F()
112 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {0, 2, 1, 4, 3})); in TEST_F()
114 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {1, 3, 0, 2, 4})); in TEST_F()
116 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {3, 0, 1, 2, 4})); in TEST_F()
118 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {3, 2, 0, 1, 4})); in TEST_F()
121 EXPECT_TRUE(internal::NonSingletonDimensionsAlign({1, 1, 2, 1, 2, 2}, in TEST_F()
[all …]
Dtranspose_functor.h130 inline bool NonSingletonDimensionsAlign(const TensorShape& input_shape, in NonSingletonDimensionsAlign() function
Dtranspose_op.cc189 } else if (!IsConjugate() && internal::NonSingletonDimensionsAlign( in Compute()