Searched refs:ExpectInferredReduceShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 61 void ExpectInferredReduceShape(const Shape& expected_inferred_shape, in ExpectInferredReduceShape() function in xla::__anon8d0540a50111::ReduceShapeInferenceTest
669 ExpectInferredReduceShape(f32_, ShapeUtil::MakeShape(F32, {128}), in TEST_F()
674 ExpectInferredReduceShape(ShapeUtil::MakeShape(F32, {3, 4}), in TEST_F()
680 ExpectInferredReduceShape(ShapeUtil::MakeShape(F32, {2, 4}), in TEST_F()
686 ExpectInferredReduceShape(ShapeUtil::MakeShape(F32, {4}), in TEST_F()
692 ExpectInferredReduceShape(ShapeUtil::MakeShape(F32, {2}), in TEST_F()
698 ExpectInferredReduceShape(ShapeUtil::MakeShape(F32, {3}), in TEST_F()
703 ExpectInferredReduceShape(ShapeUtil::MakeShape(F32, {3}), in TEST_F()
709 ExpectInferredReduceShape(f32_, ShapeUtil::MakeShape(F32, {2, 3, 4}), in TEST_F()
|