Home
last modified time | relevance | path

Searched refs:ReduceShapeInferenceTest (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference_test.cc60 class ReduceShapeInferenceTest : public ShapeInferenceTest { class
1080 TEST_F(ReduceShapeInferenceTest, ReduceVectorToScalar) { in TEST_F() argument
1085 TEST_F(ReduceShapeInferenceTest, ReduceCubeAmongFirstDimension) { in TEST_F() argument
1091 TEST_F(ReduceShapeInferenceTest, ReduceCubeAmongMiddleDimension) { in TEST_F() argument
1097 TEST_F(ReduceShapeInferenceTest, ReduceCubeAmongFirstTwoDimensions) { in TEST_F() argument
1103 TEST_F(ReduceShapeInferenceTest, ReduceCubeAmongLastTwoDimensions) { in TEST_F() argument
1109 TEST_F(ReduceShapeInferenceTest, ReduceCubeAmongFirstAndLastDimensions) { in TEST_F() argument
1120 TEST_F(ReduceShapeInferenceTest, ReduceCubeAmongAllDimensions) { in TEST_F() argument
1125 TEST_F(ReduceShapeInferenceTest, ReduceMultiOutput) { in TEST_F() argument
1137 TEST_F(ReduceShapeInferenceTest, ReduceWindowMultiOutput) { in TEST_F() argument
[all …]