Searched defs:ReshapeTest (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reshape_test.cc | 44 class ReshapeTest : public ::testing::WithParamInterface<bool>, class
47 ReshapeTest() { set_use_bfloat16(GetParam()); } in ReshapeTest() function in xla::__anon934cbaa60111::ReshapeTest
53 XLA_TEST_P(ReshapeTest, CollapseTrivial1x1) { in XLA_TEST_P() argument
69 XLA_TEST_P(ReshapeTest, CollapseTrivialR1EmptyDims) { in XLA_TEST_P() argument
83 XLA_TEST_P(ReshapeTest, CollapseTrivialR1OnlyDim) { in XLA_TEST_P() argument
98 XLA_TEST_P(ReshapeTest, SingleElementArrayToScalar) { in XLA_TEST_P() argument
116 XLA_TEST_P(ReshapeTest, ScalarToSingleElementArray) { in XLA_TEST_P() argument
132 XLA_TEST_P(ReshapeTest, Trivial0x3) { in XLA_TEST_P() argument
146 XLA_TEST_P(ReshapeTest, Trivial0x3WithParameter) { in XLA_TEST_P() argument
161 XLA_TEST_P(ReshapeTest, Trivial3x0) { in XLA_TEST_P() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | reshape_op_test.py | 26 class ReshapeTest(xla_test.XLATestCase, parameterized.TestCase): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | reshape_test_util.cc | 30 absl::Status ReshapeTest(TestExecutionEnvironment* env) { in ReshapeTest() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| H A D | reshape_transpose_test.py | 25 class ReshapeTest(trt_test.TfTrtIntegrationTestBase): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| H A D | reshape_op_test.py | 31 class ReshapeTest(test.TestCase): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_dimension_inference_test.cc | 570 TEST_F(DynamicDimensionInferenceTest, ReshapeTest) { in TEST_F() argument
|