Searched refs:ReshapeTest (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reshape_test.cc | 45 class ReshapeTest : public ::testing::WithParamInterface<bool>, class
48 ReshapeTest() { set_use_bfloat16(GetParam()); } in ReshapeTest() function in xla::__anonf687c8b00111::ReshapeTest
54 XLA_TEST_P(ReshapeTest, CollapseTrivial1x1) { in XLA_TEST_P() argument
69 XLA_TEST_P(ReshapeTest, CollapseTrivialR1EmptyDims) { in XLA_TEST_P() argument
82 XLA_TEST_P(ReshapeTest, CollapseTrivialR1OnlyDim) { in XLA_TEST_P() argument
96 XLA_TEST_P(ReshapeTest, SingleElementArrayToScalar) { in XLA_TEST_P() argument
113 XLA_TEST_P(ReshapeTest, ScalarToSingleElementArray) { in XLA_TEST_P() argument
128 XLA_TEST_P(ReshapeTest, Trivial0x3) { in XLA_TEST_P() argument
141 XLA_TEST_P(ReshapeTest, Trivial0x3WithParameter) { in XLA_TEST_P() argument
155 XLA_TEST_P(ReshapeTest, Trivial3x0) { in XLA_TEST_P() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reshape_op_test.py | 30 class ReshapeTest(xla_test.XLATestCase, parameterized.TestCase): class
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | reshape_transpose_test.py | 29 class ReshapeTest(trt_test.TfTrtIntegrationTestBase): class
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reshape_op_test.py | 31 class ReshapeTest(test.TestCase): class
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 374 TEST_F(DynamicDimensionInferenceTest, ReshapeTest) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops_test.py | 239 class ReshapeTest(Base): class
|