}
}
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
+ }
+}
+op {
name: "BatchCholesky"
input_arg {
name: "input"