Searched defs:reduce_shape (Results 1 – 5 of 5) sorted by relevance
74 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local191 auto reduce_shape = ShapeUtil::MakeShape(F32, {3, 5}); in TEST_F() local
181 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local213 auto reduce_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
3134 Shape reduce_shape = ShapeUtil::FilterDimensions( in HandleReduceWindow() local
1737 Shape reduce_shape = ShapeUtil::MakeShape(F32, {kParamLength}); in TEST_F() local
115 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local