| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference_test.cc | 38 class ShapeInferenceTest : public ::testing::Test { class
107 TEST_F(ShapeInferenceTest, UnaryNegateMatrix) { in TEST_F() argument
115 TEST_F(ShapeInferenceTest, SelectScalarPredBetweenTuples) { in TEST_F() argument
124 TEST_F(ShapeInferenceTest, SelectScalarPredBetweenArrays) { in TEST_F() argument
133 TEST_F(ShapeInferenceTest, SelectArrayPredBetweenArrays) { in TEST_F() argument
141 TEST_F(ShapeInferenceTest, SelectBadShapes) { in TEST_F() argument
172 TEST_F(ShapeInferenceTest, ClampAllMatrix) { in TEST_F() argument
179 TEST_F(ShapeInferenceTest, ClampAllScalar) { in TEST_F() argument
186 TEST_F(ShapeInferenceTest, ClampMinScalar) { in TEST_F() argument
194 TEST_F(ShapeInferenceTest, ClampMaxScalar) { in TEST_F() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | shape_inference_test.cc | 56 class ShapeInferenceTest : public ::testing::Test { class
81 TEST_F(ShapeInferenceTest, InputOutputByName) { in TEST_F() argument
128 TEST_F(ShapeInferenceTest, DimensionOrConstant) { in TEST_F() argument
141 TEST_F(ShapeInferenceTest, Run) { in TEST_F() argument
177 TEST_F(ShapeInferenceTest, AttachContext) { in TEST_F() argument
280 TEST_F(ShapeInferenceTest, RankAndDimInspection) { in TEST_F() argument
321 TEST_F(ShapeInferenceTest, NumElements) { in TEST_F() argument
335 TEST_F(ShapeInferenceTest, WithRank) { in TEST_F() argument
376 TEST_F(ShapeInferenceTest, WithRankAtMost) { in TEST_F() argument
415 TEST_F(ShapeInferenceTest, WithRankAtLeast) { in TEST_F() argument
[all …]
|
| H A D | shape_inference.h | 145 friend class ShapeInferenceTest; variable
183 friend class ShapeInferenceTest; variable
759 friend class ShapeInferenceTest; // For testing Relax functions. variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_experimental_test.cc | 88 class ShapeInferenceTest : public ::testing::Test { class
90 ShapeInferenceTest() in ShapeInferenceTest() function in tensorflow::__anonb65ec5d90111::ShapeInferenceTest
164 TEST_F(ShapeInferenceTest, InfersShapesFromInputShapes) { in TEST_F() argument
198 TEST_F(ShapeInferenceTest, InfersShapesFromInputTensors) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/ |
| H A D | shape_inference_utils_test.cc | 63 class ShapeInferenceTest : public ::testing::Test { class
67 ShapeInferenceTest() { in ShapeInferenceTest() function in mlir::tfg::ShapeInferenceTest
108 TEST_F(ShapeInferenceTest, TestShapeAndTypeInference) { in TEST_F() argument
199 TEST_F(ShapeInferenceTest, TestInferenceFailure) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/io_ops/ |
| H A D | save_restore_ops_test.py | 53 class ShapeInferenceTest(test.TestCase): class
|