// indicate there are more values, then this is representing a splat tensor.
  // We can create an MLIR Attribute more efficiently in this case.
  TensorShape input_tensor_shape(input_tensor.tensor_shape());
  if (NumberOfMaterializedElements(input_tensor) == 1 &&
      input_tensor_shape.num_elements() > 1) {
    // We first convert this TensorProto to one of shape [1]. We then create an
    // Attribute for that proto, and finally splat the Attribute.

    auto input_tensor_shape =
        mlir::cast<RankedTensorType>(input_tensor.getType());
    for (auto i = 0; i < input_tensor_shape.getShape().size(); ++i) {
      DCOMMENT("Input Tensor Shape " << i << "th is "
                                     << input_tensor_shape.getShape()[i]);
      input_tensor_dims_vec.push_back(input_tensor_shape.getShape()[i]);
    }