Home
last modified time | relevance | path

Searched refs:TransposeTest (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_test.cc67 TEST(TransposeTest, TestRefOps1D) { in TEST() argument
74 TEST(TransposeTest, TestRefOps2D) { in TEST() argument
84 TEST(TransposeTest, TestRefOps3D) { in TEST() argument
133 TEST(TransposeTest, TestRefOps3D_OneInDimension) { in TEST() argument
161 TEST(TransposeTest, TestRefOps4D) { in TEST() argument
206 TEST(TransposeTest, TestRefOps4DInt8) { TransposeTestTestRefOps4D<int8_t>(); } in TEST() argument
208 TEST(TransposeTest, TestRefOps4DInt16) { TransposeTestTestRefOps4D<int16_t>(); } in TEST() argument
270 TEST(TransposeTest, TestUnequalPermSize) { in TEST() argument
274 TEST(TransposeTest, TestPermOutOfBounds) { in TEST() argument
282 TEST(TransposeTest, Test1DInputConstTensor) { in TEST() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtranspose_test.cc31 class TransposeTest : public ClientLibraryTestBase { class
39 XLA_TEST_F(TransposeTest, Transpose0x0) { in XLA_TEST_F() argument
47 XLA_TEST_F(TransposeTest, Transpose0x42) { in XLA_TEST_F() argument
55 XLA_TEST_F(TransposeTest, Transpose7x0) { in XLA_TEST_F() argument
63 TEST_F(TransposeTest, Transpose2x2) { in TEST_F() argument
76 XLA_TEST_F(TransposeTest, Transpose0x2x3_2x3x0) { in XLA_TEST_F() argument
85 TEST_F(TransposeTest, Transpose1x2x3_2x3x1) { in TEST_F() argument
96 TEST_F(TransposeTest, Transpose1x2x3_3x2x1) { in TEST_F() argument
107 TEST_F(TransposeTest, Transpose1x2x3_1x2x3) { in TEST_F() argument
118 TEST_F(TransposeTest, MultiTranspose3x2) { in TEST_F() argument
[all …]
/external/libgav1/src/dsp/arm/
Dcommon_neon_test.cc33 class TransposeTest : public testing::Test { class
35 TransposeTest() { in TransposeTest() function in libgav1::dsp::__anon5a794cb20111::TransposeTest
44 TransposeTest(const TransposeTest&) = delete;
45 TransposeTest& operator=(const TransposeTest&) = delete;
46 ~TransposeTest() override = default;
53 using TransposeTestLowBitdepth = TransposeTest<uint8_t>;
112 using TransposeTestHighBitdepth = TransposeTest<uint16_t>;
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtranspose_test.cc44 TEST(TransposeTest, RemoveTrivialDimensions) { in TEST() argument
68 TEST(TransposeTest, CoalesceDimensions) { in TEST() argument
118 TEST(TransposeTest, InvalidTilings) { in TEST() argument
352 class TransposeTest : public ::testing::TestWithParam<TransposeTestCase> { class
387 TEST_P(TransposeTest, TransposeInt8) { TestTranspose<int8_t>(1); } in TEST_P() argument
388 TEST_P(TransposeTest, TransposeInt16) { TestTranspose<int16_t>(1); } in TEST_P() argument
389 TEST_P(TransposeTest, TransposeInt32) { TestTranspose<int32_t>(1); } in TEST_P() argument
390 TEST_P(TransposeTest, TransposeInt64) { TestTranspose<int64_t>(1); } in TEST_P() argument
391 TEST_P(TransposeTest, TransposeInt128) { TestTranspose<absl::int128>(1); } in TEST_P() argument
393 TEST_P(TransposeTest, ParallelTransposeInt8) { TestTranspose<int8_t>(16); } in TEST_P() argument
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dreshape_transpose_test.py77 class TransposeTest(trt_test.TfTrtIntegrationTestBase): class
106 class IncompatibleTransposeTest(TransposeTest):
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dtranspose_test_util.h25 absl::Status TransposeTest(TestExecutionEnvironment* env);
Dtranspose_test_util.cc115 absl::Status TransposeTest(TestExecutionEnvironment* env) { in TransposeTest() function
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dtranspose_test.cc31 auto status = TransposeTest(&exec_env_); in TEST_F()
/external/armnn/delegate/test/
DTransposeTest.cpp29 TransposeTest<float>(backends, in TransposeFP32Test()
DTransposeTestHelper.hpp87 void TransposeTest(std::vector<armnn::BackendId>& backends, in TransposeTest() function
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtranspose_test.mm37 auto status = TransposeTest(&exec_env_);
/external/armnn/delegate/
DCMakeLists.txt201 test/TransposeTest.cpp
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dtranspose_op_test.py33 class TransposeTest(test.TestCase): class
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference_test.cc492 TEST_F(DynamicDimensionInferenceTest, TransposeTest) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dconst-fold.mlir499 // A test case adopted from TransposeTest.Test3DInputConstTensor in
/external/armnn/
DLICENSE.spdx3372 FileName: ./delegate/src/test/TransposeTest.cpp