Searched refs:ShapeInferenceTest (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 34 class ShapeInferenceTest : public ::testing::Test { class
56 class ReduceShapeInferenceTest : public ShapeInferenceTest {
74 class SelectAndScatterShapeInferenceTest : public ShapeInferenceTest {
103 TEST_F(ShapeInferenceTest, UnaryNegateMatrix) { in TEST_F() argument
111 TEST_F(ShapeInferenceTest, SelectScalarPredBetweenTuples) { in TEST_F() argument
119 TEST_F(ShapeInferenceTest, SelectScalarPredBetweenArrays) { in TEST_F() argument
126 TEST_F(ShapeInferenceTest, SelectArrayPredBetweenArrays) { in TEST_F() argument
134 TEST_F(ShapeInferenceTest, SelectBadShapes) { in TEST_F() argument
164 TEST_F(ShapeInferenceTest, ClampAllMatrix) { in TEST_F() argument
171 TEST_F(ShapeInferenceTest, ClampAllScalar) { in TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 49 class ShapeInferenceTest : public ::testing::Test { class
72 TEST_F(ShapeInferenceTest, InputOutputByName) { in TEST_F() argument
115 TEST_F(ShapeInferenceTest, DimensionOrConstant) { in TEST_F() argument
128 TEST_F(ShapeInferenceTest, Run) { in TEST_F() argument
164 TEST_F(ShapeInferenceTest, AttachContext) { in TEST_F() argument
255 TEST_F(ShapeInferenceTest, RankAndDimInspection) { in TEST_F() argument
296 TEST_F(ShapeInferenceTest, NumElements) { in TEST_F() argument
310 TEST_F(ShapeInferenceTest, WithRank) { in TEST_F() argument
349 TEST_F(ShapeInferenceTest, WithRankAtMost) { in TEST_F() argument
387 TEST_F(ShapeInferenceTest, WithRankAtLeast) { in TEST_F() argument
[all …]
|
| D | shape_inference.h | 72 friend class ShapeInferenceTest; variable
113 friend class ShapeInferenceTest; variable
686 friend class ShapeInferenceTest; // For testing Relax functions. variable
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_test.cc | 34 TEST(ShapeInferenceTest, Basics) { in TEST() argument
62 TEST(ShapeInferenceTest, WhileLoop) { in TEST() argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | save_restore_ops_test.py | 56 class ShapeInferenceTest(test.TestCase): class
|