Searched defs:total_batch (Results 1 – 4 of 4) sorted by relevance
28 … int32_t device_id, bool send_epoch_end, int32_t total_batch, bool create_data_info_queue) in DataQueueNode()132 int32_t total_batch = json_obj["total_batch"]; in from_json() local
101 int32_t total_batch, bool create_data_info_queue) in DataQueueOp()1114 int64_t total_batch = 0; in SendDataToCPU() local
312 int32_t total_batch, bool create_data_info_queue) { in __anon10ea60a02602()
1599 def check_total_batch(total_batch): argument