op { | |
name: "SparseReshape" | |
input_arg { | |
name: "input_indices" | |
type: DT_INT64 | |
} | |
input_arg { | |
name: "input_shape" | |
type: DT_INT64 | |
} | |
input_arg { | |
name: "new_shape" | |
type: DT_INT64 | |
} | |
output_arg { | |
name: "output_indices" | |
type: DT_INT64 | |
} | |
output_arg { | |
name: "output_shape" | |
type: DT_INT64 | |
} | |
} |