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