bool report_error = true);

// Parses the sharding attr. This sharding attr can be binary (serialized)
// or human readable.
void EncodeSharding(mlir::Operation* op, llvm::StringRef shard_str);

// Parses "input_sharding_configuration" attribute and returns a list where i-th
// element is a list of mlir::Value's which represent inputs for the TPU