| op { |
| name: "Batch" |
| input_arg { |
| name: "in_tensors" |
| type_list_attr: "T" |
| } |
| output_arg { |
| name: "batched_tensors" |
| type_list_attr: "T" |
| } |
| output_arg { |
| name: "batch_index" |
| type: DT_INT64 |
| } |
| output_arg { |
| name: "id" |
| type: DT_INT64 |
| } |
| attr { |
| name: "num_batch_threads" |
| type: "int" |
| } |
| attr { |
| name: "max_batch_size" |
| type: "int" |
| } |
| attr { |
| name: "batch_timeout_micros" |
| type: "int" |
| } |
| attr { |
| name: "allowed_batch_sizes" |
| type: "list(int)" |
| default_value { |
| list { |
| } |
| } |
| } |
| attr { |
| name: "grad_timeout_micros" |
| type: "int" |
| } |
| 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: "T" |
| type: "list(type)" |
| has_minimum: true |
| minimum: 1 |
| } |
| } |
| op { |
| name: "Batch" |
| input_arg { |
| name: "in_tensors" |
| type_list_attr: "T" |
| } |
| output_arg { |
| name: "batched_tensors" |
| type_list_attr: "T" |
| } |
| output_arg { |
| name: "batch_index" |
| type: DT_INT64 |
| } |
| output_arg { |
| name: "id" |
| type: DT_INT64 |
| } |
| attr { |
| name: "num_batch_threads" |
| type: "int" |
| } |
| attr { |
| name: "max_batch_size" |
| type: "int" |
| } |
| attr { |
| name: "max_enqueued_batches" |
| type: "int" |
| default_value { |
| i: 10 |
| } |
| } |
| attr { |
| name: "batch_timeout_micros" |
| type: "int" |
| } |
| attr { |
| name: "allowed_batch_sizes" |
| type: "list(int)" |
| default_value { |
| list { |
| } |
| } |
| } |
| attr { |
| name: "grad_timeout_micros" |
| type: "int" |
| } |
| 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: "T" |
| type: "list(type)" |
| has_minimum: true |
| minimum: 1 |
| } |
| } |