| op { |
| name: "BatchFunction" |
| input_arg { |
| name: "in_tensors" |
| type_list_attr: "Tin" |
| } |
| input_arg { |
| name: "captured_tensors" |
| type_list_attr: "Tcaptured" |
| } |
| output_arg { |
| name: "out_tensors" |
| type_list_attr: "Tout" |
| } |
| attr { |
| name: "f" |
| type: "func" |
| } |
| attr { |
| name: "num_batch_threads" |
| type: "int" |
| } |
| attr { |
| name: "max_batch_size" |
| type: "int" |
| } |
| attr { |
| name: "batch_timeout_micros" |
| type: "int" |
| } |
| attr { |
| name: "max_enqueued_batches" |
| type: "int" |
| default_value { |
| i: 10 |
| } |
| } |
| attr { |
| name: "allowed_batch_sizes" |
| type: "list(int)" |
| default_value { |
| list { |
| } |
| } |
| } |
| attr { |
| name: "container" |
| type: "string" |
| default_value { |
| s: "" |
| } |
| } |
| attr { |
| name: "shared_name" |
| type: "string" |
| default_value { |
| s: "" |
| } |
| } |
| attr { |
| name: "batching_queue" |
| type: "string" |
| default_value { |
| s: "" |
| } |
| } |
| attr { |
| name: "Tin" |
| type: "list(type)" |
| has_minimum: true |
| minimum: 1 |
| } |
| attr { |
| name: "Tcaptured" |
| type: "list(type)" |
| has_minimum: true |
| } |
| attr { |
| name: "Tout" |
| type: "list(type)" |
| has_minimum: true |
| minimum: 1 |
| } |
| } |