op { | |
name: "LRNGrad" | |
input_arg { | |
name: "input_grads" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "input_image" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "output_image" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "output" | |
type_attr: "T" | |
} | |
attr { | |
name: "depth_radius" | |
type: "int" | |
default_value { | |
i: 5 | |
} | |
} | |
attr { | |
name: "bias" | |
type: "float" | |
default_value { | |
f: 1 | |
} | |
} | |
attr { | |
name: "alpha" | |
type: "float" | |
default_value { | |
f: 1 | |
} | |
} | |
attr { | |
name: "beta" | |
type: "float" | |
default_value { | |
f: 0.5 | |
} | |
} | |
attr { | |
name: "T" | |
type: "type" | |
default_value { | |
type: DT_FLOAT | |
} | |
allowed_values { | |
list { | |
type: DT_FLOAT | |
type: DT_HALF | |
} | |
} | |
} | |
} | |
op { | |
name: "LRNGrad" | |
input_arg { | |
name: "input_grads" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "input_image" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "output_image" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "output" | |
type_attr: "T" | |
} | |
attr { | |
name: "depth_radius" | |
type: "int" | |
default_value { | |
i: 5 | |
} | |
} | |
attr { | |
name: "bias" | |
type: "float" | |
default_value { | |
f: 1 | |
} | |
} | |
attr { | |
name: "alpha" | |
type: "float" | |
default_value { | |
f: 1 | |
} | |
} | |
attr { | |
name: "beta" | |
type: "float" | |
default_value { | |
f: 0.5 | |
} | |
} | |
attr { | |
name: "T" | |
type: "type" | |
default_value { | |
type: DT_FLOAT | |
} | |
allowed_values { | |
list { | |
type: DT_HALF | |
type: DT_BFLOAT16 | |
type: DT_FLOAT | |
} | |
} | |
} | |
} |