Home
last modified time | relevance | path

Searched refs:AddReshardOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.h75 Status AddReshardOp(Graph* graph, const HostTrainingLoopInfo& host_loop_info);
Dhost_training_loop_optimization_util.cc440 Status AddReshardOp(Graph* graph, const HostTrainingLoopInfo& host_loop_info) { in AddReshardOp() function
Ddistributed_tpu_rewrite_pass.cc4408 TF_RETURN_IF_ERROR(tpu::AddReshardOp(function_graph, host_loop)); in PerformHostTrainingLoopOptimization()
4412 TF_RETURN_IF_ERROR(tpu::AddReshardOp(graph, host_loop)); in PerformHostTrainingLoopOptimization()