Home
last modified time | relevance | path

Searched defs:ShapeInferenceTest (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc36 class ShapeInferenceTest : public ::testing::Test { class
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
184 TEST_F(ShapeInferenceTest, ClampMinScalar) { in TEST_F() argument
192 TEST_F(ShapeInferenceTest, ClampMaxScalar) { in TEST_F() argument
[all …]
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc58 class ShapeInferenceTest : public ::testing::Test { class
81 TEST_F(ShapeInferenceTest, InputOutputByName) { in TEST_F() argument
124 TEST_F(ShapeInferenceTest, DimensionOrConstant) { in TEST_F() argument
137 TEST_F(ShapeInferenceTest, Run) { in TEST_F() argument
172 TEST_F(ShapeInferenceTest, AttachContext) { in TEST_F() argument
273 TEST_F(ShapeInferenceTest, RankAndDimInspection) { in TEST_F() argument
314 TEST_F(ShapeInferenceTest, NumElements) { in TEST_F() argument
328 TEST_F(ShapeInferenceTest, WithRank) { in TEST_F() argument
368 TEST_F(ShapeInferenceTest, WithRankAtMost) { in TEST_F() argument
407 TEST_F(ShapeInferenceTest, WithRankAtLeast) { in TEST_F() argument
[all …]
Dshape_inference.h71 friend class ShapeInferenceTest; variable
109 friend class ShapeInferenceTest; variable
683 friend class ShapeInferenceTest; // For testing Relax functions. variable
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc88 class ShapeInferenceTest : public ::testing::Test { class
90 ShapeInferenceTest() in ShapeInferenceTest() function in tensorflow::__anon7f1c240f0111::ShapeInferenceTest
164 TEST_F(ShapeInferenceTest, InfersShapesFromInputShapes) { in TEST_F() argument
198 TEST_F(ShapeInferenceTest, InfersShapesFromInputTensors) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc36 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/
Dsave_restore_ops_test.py57 class ShapeInferenceTest(test.TestCase): class