HomeSort by relevance Sort by last modified time
    Searched refs:AutoParallel (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/grappler/optimizers/
auto_parallel.h 27 class AutoParallel : public GraphOptimizer {
29 AutoParallel(int num_replicas) : num_replicas_(num_replicas) {
32 ~AutoParallel() override {}
34 string name() const override { return "autoparallel"; };
auto_parallel.cc 32 const char kAutoParallelPrefix[] = "AutoParallel";
34 NodeDef* AutoParallel::AddNodeDivConst() {
51 NodeDef* AutoParallel::AddNodeDiv(const string& name, const string& input_a,
64 NodeDef* AutoParallel::AddNodeControl(const string& name,
76 Status AutoParallel::Initialize(const GrapplerItem& item) {
200 bool AutoParallel::NotSharedNode(const string& name) {
204 void AutoParallel::AddSharedNodes(GraphDef* graph) {
218 void AutoParallel::AddOneReplica(GraphDef* graph, int number) {
238 void AutoParallel::BuildGraph(GraphDef* graph) {
263 Status AutoParallel::Optimize(Cluster* cluster, const GrapplerItem& item
    [all...]
meta_optimizer.cc 71 if (optimizer == "autoparallel") {
73 new AutoParallel(cfg_.auto_parallel().num_replicas()));
126 new AutoParallel(cfg_.auto_parallel().num_replicas())));
131 "autoparallel", "arithmetic", "dependency", "loop"};
auto_parallel_test.cc 51 AutoParallel parallel(2);
59 EXPECT_EQ("AutoParallel-Replica-0/constant_a", node_assign.input(1));
75 EXPECT_EQ("AutoParallel-Replica-0/AutoParallel-Div-Const",
79 EXPECT_EQ("AutoParallel-Replica-0/AutoParallel-Div-apply_gradient",
82 EXPECT_EQ("AutoParallel-Replica-0/add", node_add0.name());
85 EXPECT_EQ("AutoParallel-Replica-0/apply_gradient", node_gradient0.name());
88 EXPECT_EQ("AutoParallel-Replica-0/constant_a", node_constant_a0.name());
91 EXPECT_EQ("AutoParallel-Replica-0/dequeue", node_dequeue0.name())
    [all...]

Completed in 157 milliseconds