Searched refs:host_loop_info (Results 1 – 2 of 2) sorted by relevance
75 Status AddReshardOp(Graph* graph, const HostTrainingLoopInfo& host_loop_info);
440 Status AddReshardOp(Graph* graph, const HostTrainingLoopInfo& host_loop_info) { in AddReshardOp() argument441 const auto& compile_node_name = host_loop_info.compile_node_name; in AddReshardOp()453 ExtractExecuteNodeInfo(compile_node, *graph, host_loop_info.loop_nodes, in AddReshardOp()478 const auto& cond_node_name = host_loop_info.loop_cond_node_name; in AddReshardOp()