blob: 5e2ea7aca10fa713995553139ddb2580adf66253 [file] [log] [blame]
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
}
}