op { | |
name: "MapPeek" | |
input_arg { | |
name: "key" | |
type: DT_INT64 | |
} | |
input_arg { | |
name: "indices" | |
type: DT_INT32 | |
} | |
output_arg { | |
name: "values" | |
type_list_attr: "dtypes" | |
} | |
attr { | |
name: "capacity" | |
type: "int" | |
default_value { | |
i: 0 | |
} | |
has_minimum: true | |
} | |
attr { | |
name: "memory_limit" | |
type: "int" | |
default_value { | |
i: 0 | |
} | |
has_minimum: true | |
} | |
attr { | |
name: "dtypes" | |
type: "list(type)" | |
has_minimum: true | |
minimum: 1 | |
} | |
attr { | |
name: "container" | |
type: "string" | |
default_value { | |
s: "" | |
} | |
} | |
attr { | |
name: "shared_name" | |
type: "string" | |
default_value { | |
s: "" | |
} | |
} | |
is_stateful: true | |
} |