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 | |
} |