Searched refs:InferReduceShape (Results 1 – 9 of 9) sorted by relevance
67 auto inferred_status = ShapeInference::InferReduceShape( in ExpectInferredReduceShape()1095 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1134 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1146 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1158 auto inferred_status = ShapeInference::InferReduceShape({}, {0, 1}, to_apply); in TEST_F()1192 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1205 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1218 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1229 auto inferred_status = ShapeInference::InferReduceShape( in TEST_F()1241 ShapeInference::InferReduceShape({&arg_shape, &f32_}, in TEST_F()[all …]
159 static StatusOr<Shape> InferReduceShape(
671 CheckShape(reduce, ShapeInference::InferReduceShape( in HandleReduce()
2415 ShapeInference::InferReduceShape( in HandleReduce()
2113 /* static */ StatusOr<Shape> ShapeInference::InferReduceShape( in InferReduceShape() function in xla::ShapeInference
1836 return ShapeInference::InferReduceShape( in ParseInstructionRhs()
3297 ShapeInference::InferReduceShape( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
2743 ShapeInference::InferReduceShape(new_operand_shapes, hlo->dimensions(), in HandleReduce()
2453 ShapeInference::InferReduceShape( in Reduce()