blob: 1f96da6f7886e69c25670c0d08333ed76a6dba1f [file] [log] [blame]
op {
name: "ParameterizedTruncatedNormal"
input_arg {
name: "shape"
type_attr: "T"
}
input_arg {
name: "means"
type_attr: "dtype"
}
input_arg {
name: "stdevs"
type_attr: "dtype"
}
input_arg {
name: "minvals"
type_attr: "dtype"
}
input_arg {
name: "maxvals"
type_attr: "dtype"
}
output_arg {
name: "output"
type_attr: "dtype"
}
attr {
name: "seed"
type: "int"
default_value {
i: 0
}
}
attr {
name: "seed2"
type: "int"
default_value {
i: 0
}
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
is_stateful: true
}
op {
name: "ParameterizedTruncatedNormal"
input_arg {
name: "shape"
type_attr: "T"
}
input_arg {
name: "means"
type_attr: "dtype"
}
input_arg {
name: "stdevs"
type_attr: "dtype"
}
input_arg {
name: "minvals"
type_attr: "dtype"
}
input_arg {
name: "maxvals"
type_attr: "dtype"
}
output_arg {
name: "output"
type_attr: "dtype"
}
attr {
name: "seed"
type: "int"
default_value {
i: 0
}
}
attr {
name: "seed2"
type: "int"
default_value {
i: 0
}
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
is_stateful: true
}