op { | |
name: "MaxPool3DGradGrad" | |
input_arg { | |
name: "orig_input" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "orig_output" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "grad" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "output" | |
type_attr: "T" | |
} | |
attr { | |
name: "ksize" | |
type: "list(int)" | |
has_minimum: true | |
minimum: 5 | |
} | |
attr { | |
name: "strides" | |
type: "list(int)" | |
has_minimum: true | |
minimum: 5 | |
} | |
attr { | |
name: "padding" | |
type: "string" | |
allowed_values { | |
list { | |
s: "SAME" | |
s: "VALID" | |
} | |
} | |
} | |
attr { | |
name: "data_format" | |
type: "string" | |
default_value { | |
s: "NDHWC" | |
} | |
allowed_values { | |
list { | |
s: "NDHWC" | |
s: "NCDHW" | |
} | |
} | |
} | |
attr { | |
name: "T" | |
type: "type" | |
allowed_values { | |
list { | |
type: DT_FLOAT | |
} | |
} | |
} | |
} | |
op { | |
name: "MaxPool3DGradGrad" | |
input_arg { | |
name: "orig_input" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "orig_output" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "grad" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "output" | |
type_attr: "T" | |
} | |
attr { | |
name: "ksize" | |
type: "list(int)" | |
has_minimum: true | |
minimum: 5 | |
} | |
attr { | |
name: "strides" | |
type: "list(int)" | |
has_minimum: true | |
minimum: 5 | |
} | |
attr { | |
name: "padding" | |
type: "string" | |
allowed_values { | |
list { | |
s: "SAME" | |
s: "VALID" | |
} | |
} | |
} | |
attr { | |
name: "data_format" | |
type: "string" | |
default_value { | |
s: "NDHWC" | |
} | |
allowed_values { | |
list { | |
s: "NDHWC" | |
s: "NCDHW" | |
} | |
} | |
} | |
attr { | |
name: "T" | |
type: "type" | |
allowed_values { | |
list { | |
type: DT_FLOAT | |
type: DT_DOUBLE | |
type: DT_INT32 | |
type: DT_UINT8 | |
type: DT_INT16 | |
type: DT_INT8 | |
type: DT_INT64 | |
type: DT_BFLOAT16 | |
type: DT_UINT16 | |
type: DT_HALF | |
type: DT_UINT32 | |
type: DT_UINT64 | |
} | |
} | |
} | |
} |