Searched refs:AutoParallel (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_parallel.cc | 35 NodeDef* AutoParallel::AddNodeDivConst() { in AddNodeDivConst()
52 NodeDef* AutoParallel::AddNodeDiv(const string& name, const string& input_a, in AddNodeDiv()
65 NodeDef* AutoParallel::AddNodeControl(const string& name, in AddNodeControl()
77 Status AutoParallel::Initialize(const GrapplerItem& item) { in Initialize()
203 bool AutoParallel::NotSharedNode(const string& name) { in NotSharedNode()
207 void AutoParallel::AddSharedNodes(GraphDef* graph) { in AddSharedNodes()
221 void AutoParallel::AddOneReplica(GraphDef* graph, int number) { in AddOneReplica()
241 void AutoParallel::BuildGraph(GraphDef* graph) { in BuildGraph()
266 Status AutoParallel::Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
273 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 | 207 MK_OPT("autoparallel", new AutoParallel(cfg_.auto_parallel().num_replicas())); in MakeNewOptimizer()
309 MakeUnique<AutoParallel>(cfg_.auto_parallel().num_replicas())); in InitializeOptimizers()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | rewriter_config.proto | 175 // Configures AutoParallel optimization passes either through the
190 // Of the RewriterConfig options, only the AutoParallel configuration options
|