// Infers the output shape of XlaConvV2Op based on the input shapes
  bool InferShapeForXlaConvV2Op(XlaConvV2Op op);

  // Infers the output shape of XlaReduceWindowOp based on the input shapes.
  bool InferShapeForXlaReduceWindowOp(XlaReduceWindowOp op);

  // Infers the output shape of XlaSelectAndScatterOp based on the input shapes.
  bool InferShapeForXlaSelectAndScatterOp(XlaSelectAndScatterOp op);