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

  /external/tensorflow/tensorflow/core/kernels/
transpose_util_test.cc 103 EXPECT_TRUE(internal::NonSingletonDimensionsAlign({2, 1, 2}, {1, 0, 2}));
104 EXPECT_TRUE(internal::NonSingletonDimensionsAlign({2, 1, 2}, {0, 2, 1}));
105 EXPECT_FALSE(internal::NonSingletonDimensionsAlign({2, 1, 2}, {2, 0, 1}));
106 EXPECT_FALSE(internal::NonSingletonDimensionsAlign({2, 1, 2}, {2, 1, 0}));
110 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {0, 2, 4, 1, 3}));
112 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {0, 2, 1, 4, 3}));
114 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {1, 3, 0, 2, 4}));
116 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {3, 0, 1, 2, 4}));
118 internal::NonSingletonDimensionsAlign({2, 1, 2, 1, 2}, {3, 2, 0, 1, 4}));
121 EXPECT_TRUE(internal::NonSingletonDimensionsAlign({1, 1, 2, 1, 2, 2}
    [all...]
transpose_functor.h 130 inline bool NonSingletonDimensionsAlign(const TensorShape& input_shape,
transpose_op.cc 189 } else if (!IsConjugate() && internal::NonSingletonDimensionsAlign(

Completed in 66 milliseconds