Searched refs:InferBroadcastShape (Results 1 – 4 of 4) sorted by relevance
218 static StatusOr<Shape> InferBroadcastShape(223 static StatusOr<Shape> InferBroadcastShape(
934 auto status = ShapeInference::InferBroadcastShape(scalar_shape, {}); in TEST_F()940 auto status = ShapeInference::InferBroadcastShape(scalar_shape, {3}); in TEST_F()945 auto status = ShapeInference::InferBroadcastShape(oned_shape, {}); in TEST_F()951 auto status = ShapeInference::InferBroadcastShape(scalar_shape, {2, 3}); in TEST_F()956 auto status = ShapeInference::InferBroadcastShape(oned_shape, {2}); in TEST_F()
2607 /* static */ StatusOr<Shape> ShapeInference::InferBroadcastShape( in InferBroadcastShape() function in xla::ShapeInference2625 /* static */ StatusOr<Shape> ShapeInference::InferBroadcastShape( in InferBroadcastShape() function in xla::ShapeInference
653 ShapeInference::InferBroadcastShape(operand_shape, broadcast_sizes)); in Broadcast()690 TF_RETURN_IF_ERROR(ShapeInference::InferBroadcastShape( in BroadcastInDim()