TensorType ty =
      tensorflow::GetTypeFromTFTensorShape({size}, builder->getIntegerType(64));
  return DenseIntElementsAttr::get(ty, vals);
}

// Return an Attr representation of the value.
static DenseElementsAttr GetF32Scalar(OpBuilder *builder, float value) {
  return DenseElementsAttr::get(
      tensorflow::GetTypeFromTFTensorShape({}, builder->getF32Type()),
      FloatAttr::get(builder->getF32Type(), value));
}