Home
last modified time | relevance | path

Searched refs:TransposeArray2D (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dtranspose_test.cc141 auto expected = ReferenceUtil::TransposeArray2D(*aoperand); in TEST_F()
153 auto expected = ReferenceUtil::TransposeArray2D(*aoperand); in TEST_F()
H A Dreshape_test.cc262 auto expected = ReferenceUtil::TransposeArray2D(*simple); in XLA_TEST_P()
280 auto expected = ReferenceUtil::TransposeArray2D(*a4x3); in XLA_TEST_P()
311 auto expected = ReferenceUtil::TransposeArray2D(*a4x3); in XLA_TEST_P()
H A Ddot_operation_test.cc805 lhs = ReferenceUtil::TransposeArray2D(*lhs); in XLA_TYPED_TEST()
808 rhs = ReferenceUtil::TransposeArray2D(*rhs); in XLA_TYPED_TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util_test.cc53 TEST_F(ReferenceUtilTest, TransposeArray2D) { in TEST_F() argument
54 auto result = ReferenceUtil::TransposeArray2D(*matrix_); in TEST_F()
H A Dreference_util.h41 static std::unique_ptr<Array2D<T>> TransposeArray2D( in TransposeArray2D() function