Home
last modified time | relevance | path

Searched refs:GraphOptimizerContext (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage.h48 struct GraphOptimizerContext { struct
49 GraphOptimizerContext(const std::unordered_set<string>* nodes_to_preserve, in GraphOptimizerContext() argument
69 Status GetInputNode(const GraphOptimizerContext& ctx, const string& input, argument
71 Status GetTensorProperties(const GraphOptimizerContext& ctx,
75 NodeDef* AddCopyNode(const GraphOptimizerContext& ctx, const string& name,
77 NodeDef* AddEmptyNode(const GraphOptimizerContext& ctx, const string& name);
120 const GraphOptimizerContext& ctx) in GraphOptimizerStage()
206 const GraphOptimizerContext& ctx() const { return ctx_; } in ctx()
221 const GraphOptimizerContext ctx_;
Dgraph_optimizer_stage_test.cc41 const GraphOptimizerContext& ctx) in FakeOptimizerStage()
64 GraphOptimizerContext ctx(/*nodes_to_preserve*/ nullptr, in TEST_F()
93 GraphOptimizerContext ctx(/*nodes_to_preserve*/ nullptr, in TEST_F()
122 GraphOptimizerContext ctx(/*nodes_to_preserve*/ nullptr, in TEST_F()
156 GraphOptimizerContext ctx(/*nodes_to_preserve*/ nullptr, in TEST_F()
198 GraphOptimizerContext ctx(/*nodes_to_preserve*/ nullptr, in TEST_F()
Dgraph_optimizer_stage.cc31 Status GetInputNode(const GraphOptimizerContext& ctx, const string& input, in GetInputNode()
43 Status GetTensorProperties(const GraphOptimizerContext& ctx, in GetTensorProperties()
74 NodeDef* AddCopyNode(const GraphOptimizerContext& ctx, const string& name, in AddCopyNode()
86 NodeDef* AddEmptyNode(const GraphOptimizerContext& ctx, const string& name) { in AddEmptyNode()
Darithmetic_optimizer.cc204 const GraphOptimizerContext& ctx, in ArithmeticOptimizerStage()
278 const string& name, const GraphOptimizerContext& ctx, in ArithmeticNodesGroupOptimizerStage()
459 explicit AddOpsRewriteStage(const GraphOptimizerContext& ctx, in AddOpsRewriteStage()
677 const GraphOptimizerContext& ctx, in HoistCommonFactorOutOfAggregation()
894 explicit MinimizeBroadcasts(const GraphOptimizerContext& ctx, in MinimizeBroadcasts()
1086 explicit RemoveIdentityTranspose(const GraphOptimizerContext& ctx, in RemoveIdentityTranspose()
1193 explicit RemoveInvolution(const GraphOptimizerContext& ctx, in RemoveInvolution()
1232 const GraphOptimizerContext& ctx, in RemoveRedundantBitcastStage()
1280 explicit RemoveRedundantCastStage(const GraphOptimizerContext& ctx, in RemoveRedundantCastStage()
1305 explicit RemoveNegationStage(const GraphOptimizerContext& ctx, in RemoveNegationStage()
[all …]