Add missing attributes to the schema GivenTensorFill operators (#6330)
diff --git a/caffe2/operators/given_tensor_fill_op.cc b/caffe2/operators/given_tensor_fill_op.cc
index b1866d5..9912c0d 100644
--- a/caffe2/operators/given_tensor_fill_op.cc
+++ b/caffe2/operators/given_tensor_fill_op.cc
@@ -26,32 +26,137 @@
.NumInputs(0, 1)
.NumOutputs(1)
.AllowInplace({{0, 0}})
+ .Arg(
+ "values",
+ "The value for the elements of the output tensor.",
+ true /* required */)
+ .Arg(
+ "dtype",
+ "The data type for the elements of the output tensor."
+ "Strictly must be one of the types from DataType enum in TensorProto.")
+ .Arg(
+ "shape",
+ "The shape of the output tensor."
+ "Cannot set the shape argument and pass in an input at the same time.")
+ .Arg(
+ "extra_shape",
+ "The additional dimensions appended at the end of the shape indicated"
+ "by the input blob."
+ "Cannot set the extra_shape argument when there is no input blob.")
+ .Arg(
+ "input_as_shape",
+ "1D tensor containing the desired output shape. First input must be in CPU context.")
.TensorInferenceFunction(FillerTensorInference<>);
+
OPERATOR_SCHEMA(GivenTensorDoubleFill)
.NumInputs(0, 1)
.NumOutputs(1)
.AllowInplace({{0, 0}})
+ .Arg(
+ "values",
+ "The value for the elements of the output tensor.",
+ true /* required */)
+ .Arg(
+ "shape",
+ "The shape of the output tensor."
+ "Cannot set the shape argument and pass in an input at the same time.")
+ .Arg(
+ "extra_shape",
+ "The additional dimensions appended at the end of the shape indicated"
+ "by the input blob."
+ "Cannot set the extra_shape argument when there is no input blob.")
+ .Arg(
+ "input_as_shape",
+ "1D tensor containing the desired output shape. First input must be in CPU context.")
.TensorInferenceFunction(
FillerTensorInference<TensorProto_DataType_DOUBLE>);
+
OPERATOR_SCHEMA(GivenTensorBoolFill)
.NumInputs(0, 1)
.NumOutputs(1)
.AllowInplace({{0, 0}})
+ .Arg(
+ "values",
+ "The value for the elements of the output tensor.",
+ true /* required */)
+ .Arg(
+ "shape",
+ "The shape of the output tensor."
+ "Cannot set the shape argument and pass in an input at the same time.")
+ .Arg(
+ "extra_shape",
+ "The additional dimensions appended at the end of the shape indicated"
+ "by the input blob."
+ "Cannot set the extra_shape argument when there is no input blob.")
+ .Arg(
+ "input_as_shape",
+ "1D tensor containing the desired output shape. First input must be in CPU context.")
.TensorInferenceFunction(FillerTensorInference<TensorProto_DataType_BOOL>);
+
OPERATOR_SCHEMA(GivenTensorIntFill)
.NumInputs(0, 1)
.NumOutputs(1)
.AllowInplace({{0, 0}})
+ .Arg(
+ "values",
+ "The value for the elements of the output tensor.",
+ true /* required */)
+ .Arg(
+ "shape",
+ "The shape of the output tensor."
+ "Cannot set the shape argument and pass in an input at the same time.")
+ .Arg(
+ "extra_shape",
+ "The additional dimensions appended at the end of the shape indicated"
+ "by the input blob."
+ "Cannot set the extra_shape argument when there is no input blob.")
+ .Arg(
+ "input_as_shape",
+ "1D tensor containing the desired output shape. First input must be in CPU context.")
.TensorInferenceFunction(FillerTensorInference<TensorProto_DataType_INT32>);
+
OPERATOR_SCHEMA(GivenTensorInt64Fill)
.NumInputs(0, 1)
.NumOutputs(1)
.AllowInplace({{0, 0}})
+ .Arg(
+ "values",
+ "The value for the elements of the output tensor.",
+ true /* required */)
+ .Arg(
+ "shape",
+ "The shape of the output tensor."
+ "Cannot set the shape argument and pass in an input at the same time.")
+ .Arg(
+ "extra_shape",
+ "The additional dimensions appended at the end of the shape indicated"
+ "by the input blob."
+ "Cannot set the extra_shape argument when there is no input blob.")
+ .Arg(
+ "input_as_shape",
+ "1D tensor containing the desired output shape. First input must be in CPU context.")
.TensorInferenceFunction(FillerTensorInference<TensorProto_DataType_INT64>);
+
OPERATOR_SCHEMA(GivenTensorStringFill)
.NumInputs(0, 1)
.NumOutputs(1)
.AllowInplace({{0, 0}})
+ .Arg(
+ "values",
+ "The value for the elements of the output tensor.",
+ true /* required */)
+ .Arg(
+ "shape",
+ "The shape of the output tensor."
+ "Cannot set the shape argument and pass in an input at the same time.")
+ .Arg(
+ "extra_shape",
+ "The additional dimensions appended at the end of the shape indicated"
+ "by the input blob."
+ "Cannot set the extra_shape argument when there is no input blob.")
+ .Arg(
+ "input_as_shape",
+ "1D tensor containing the desired output shape. First input must be in CPU context.")
.TensorInferenceFunction(
FillerTensorInference<TensorProto_DataType_STRING>);
