op { | |
name: "UnbatchGrad" | |
input_arg { | |
name: "original_input" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "batch_index" | |
type: DT_INT64 | |
} | |
input_arg { | |
name: "grad" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "id" | |
type: DT_INT64 | |
} | |
output_arg { | |
name: "batched_grad" | |
type_attr: "T" | |
} | |
attr { | |
name: "container" | |
type: "string" | |
default_value { | |
s: "" | |
} | |
} | |
attr { | |
name: "shared_name" | |
type: "string" | |
default_value { | |
s: "" | |
} | |
} | |
attr { | |
name: "T" | |
type: "type" | |
} | |
} |