Searched refs:ShapeInferenceTest (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 36 class ShapeInferenceTest : public ::testing::Test { class
58 class ReduceShapeInferenceTest : public ShapeInferenceTest {
76 class SelectAndScatterShapeInferenceTest : public ShapeInferenceTest {
105 TEST_F(ShapeInferenceTest, UnaryNegateMatrix) { in TEST_F() argument
113 TEST_F(ShapeInferenceTest, SelectScalarPredBetweenTuples) { in TEST_F() argument
122 TEST_F(ShapeInferenceTest, SelectScalarPredBetweenArrays) { in TEST_F() argument
131 TEST_F(ShapeInferenceTest, SelectArrayPredBetweenArrays) { in TEST_F() argument
139 TEST_F(ShapeInferenceTest, SelectBadShapes) { in TEST_F() argument
170 TEST_F(ShapeInferenceTest, ClampAllMatrix) { in TEST_F() argument
177 TEST_F(ShapeInferenceTest, ClampAllScalar) { in TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 57 class ShapeInferenceTest : public ::testing::Test { class
80 TEST_F(ShapeInferenceTest, InputOutputByName) { in TEST_F() argument
123 TEST_F(ShapeInferenceTest, DimensionOrConstant) { in TEST_F() argument
136 TEST_F(ShapeInferenceTest, Run) { in TEST_F() argument
171 TEST_F(ShapeInferenceTest, AttachContext) { in TEST_F() argument
272 TEST_F(ShapeInferenceTest, RankAndDimInspection) { in TEST_F() argument
313 TEST_F(ShapeInferenceTest, NumElements) { in TEST_F() argument
327 TEST_F(ShapeInferenceTest, WithRank) { in TEST_F() argument
366 TEST_F(ShapeInferenceTest, WithRankAtMost) { in TEST_F() argument
404 TEST_F(ShapeInferenceTest, WithRankAtLeast) { in TEST_F() argument
[all …]
|
| D | shape_inference.h | 70 friend class ShapeInferenceTest; variable
108 friend class ShapeInferenceTest; variable
673 friend class ShapeInferenceTest; // For testing Relax functions. variable
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 88 class ShapeInferenceTest : public ::testing::Test { class
90 ShapeInferenceTest() in ShapeInferenceTest() function in tensorflow::__anon347214ea0111::ShapeInferenceTest
96 ~ShapeInferenceTest() override { in ~ShapeInferenceTest()
164 TEST_F(ShapeInferenceTest, InfersShapesFromInputShapes) { in TEST_F() argument
198 TEST_F(ShapeInferenceTest, InfersShapesFromInputTensors) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_test.cc | 36 TEST(ShapeInferenceTest, Basics) { in TEST() argument
64 TEST(ShapeInferenceTest, WhileLoop) { in TEST() argument
125 TEST(ShapeInferenceTest, WhileLoopWithResource) { in TEST() argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | save_restore_ops_test.py | 57 class ShapeInferenceTest(test.TestCase): class
|