Searched refs:AutoParallel (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_parallel.cc | 34 NodeDef* AutoParallel::AddNodeDivConst() { in AddNodeDivConst()
51 NodeDef* AutoParallel::AddNodeDiv(const string& name, const string& input_a, in AddNodeDiv()
64 NodeDef* AutoParallel::AddNodeControl(const string& name, in AddNodeControl()
76 Status AutoParallel::Initialize(const GrapplerItem& item) { in Initialize()
200 bool AutoParallel::NotSharedNode(const string& name) { in NotSharedNode()
204 void AutoParallel::AddSharedNodes(GraphDef* graph) { in AddSharedNodes()
218 void AutoParallel::AddOneReplica(GraphDef* graph, int number) { in AddOneReplica()
238 void AutoParallel::BuildGraph(GraphDef* graph) { in BuildGraph()
263 Status AutoParallel::Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
270 void AutoParallel::Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
|
| D | auto_parallel.h | 27 class AutoParallel : public GraphOptimizer {
29 AutoParallel(int num_replicas) : num_replicas_(num_replicas) { in AutoParallel() function
32 ~AutoParallel() override {} in ~AutoParallel()
|
| D | auto_parallel_test.cc | 51 AutoParallel parallel(2); in TEST_F()
|
| D | meta_optimizer.cc | 133 MK_OPT("autoparallel", new AutoParallel(cfg_.auto_parallel().num_replicas())); in MakeNewOptimizer()
215 MakeUnique<AutoParallel>(cfg_.auto_parallel().num_replicas())); in InitializeOptimizers()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | rewriter_config.proto | 140 // Configures AutoParallel optimization passes either through the
155 // Of the RewriterConfig options, only the AutoParallel configuration options
|