Searched defs:total_batch (Results 1 – 3 of 3) sorted by relevance
35 … int32_t device_id, bool send_epoch_end, int32_t total_batch, bool create_data_info_queue) in TransferNode()143 int32_t total_batch = json_obj["total_batch"]; in from_json() local
31 bool send_epoch_end, int32_t total_batch, bool create_data_info_queue) in DeviceQueueOp()642 int64_t total_batch = 0; in SendDataToCPU() local
284 bool create_data_info_queue) { in __anon672ce46a2402()