Searched refs:InferReduceShape (Results 1 – 9 of 9) sorted by relevance
69 auto inferred_status = ShapeInference::InferReduceShape( in ExpectInferredReduceShape()1130 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1169 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1181 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1193 auto inferred_status = ShapeInference::InferReduceShape({}, {0, 1}, to_apply); in TEST_F()1227 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1240 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1253 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1264 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1276 ShapeInference::InferReduceShape({&arg_shape, &f32_}, in TEST_F()[all …]
190 static StatusOr<Shape> InferReduceShape(
1034 CheckShape(reduce, ShapeInference::InferReduceShape( in HandleReduce()
2216 /* static */ StatusOr<Shape> ShapeInference::InferReduceShape( in InferReduceShape() function in xla::ShapeInference
3720 ShapeInference::InferReduceShape( in HandleReduce()
2204 return ShapeInference::InferReduceShape( in CreateInstruction()
4343 ShapeInference::InferReduceShape( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3515 ShapeInference::InferReduceShape(new_operand_shapes, hlo->dimensions(), in HandleReduce()
2608 ShapeInference::InferReduceShape( in Reduce()