Searched refs:ScopedAllocatorOptimizer (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.h | 30 class ScopedAllocatorOptimizer; variable
34 class ScopedAllocatorOptimizer : public GraphOptimizer {
36 ScopedAllocatorOptimizer(RewriterConfig::Toggle opt_level,
38 ~ScopedAllocatorOptimizer() override;
79 virtual Status Rewrite(ScopedAllocatorOptimizer* paopti,
|
| D | scoped_allocator_optimizer.cc | 201 void ScopedAllocatorOptimizer::ExtendNodeAttr(StringPiece name, in ExtendNodeAttr()
216 class UnaryElementwiseRewriter : public ScopedAllocatorOptimizer::Rewriter {
279 Status AnalyzeInputs(ScopedAllocatorOptimizer* sa_opti, NodeMap* node_map, in AnalyzeInputs()
314 ScopedAllocatorOptimizer* sa_opti, GraphDef* graph, NodeMap* node_map, in ConstructScopedAllocatorNode()
342 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", in ConstructScopedAllocatorNode()
568 Status Rewrite(ScopedAllocatorOptimizer* sa_opti, int64 invocation_count, in Rewrite()
655 ScopedAllocatorOptimizer::ScopedAllocatorOptimizer( in ScopedAllocatorOptimizer() function in tensorflow::grappler::ScopedAllocatorOptimizer
675 Status ScopedAllocatorOptimizer::Optimize(Cluster* /*cluster*/, in Optimize()
689 LOG_WARNING_AND_RETURN_IF_ERROR(ScopedAllocatorOptimizer::ProcessGraphDef( in Optimize()
696 ScopedAllocatorOptimizer::Rewriter* ScopedAllocatorOptimizer::GetRewriter( in GetRewriter()
[all …]
|
| D | scoped_allocator_optimizer_test.cc | 119 ScopedAllocatorOptimizer sao(RewriterConfig::ON, opts); in TEST_F()
120 ScopedAllocatorOptimizer::OpNameSet ons; in TEST_F()
224 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {0, 2}, &nd); in TEST_F()
225 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {6, 7}, &nd); in TEST_F()
226 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {2, 3}, &nd); in TEST_F()
|
| D | meta_optimizer.cc | 138 new ScopedAllocatorOptimizer(cfg_.scoped_allocator_optimization(), in MakeNewOptimizer()
218 optimizers->push_back(MakeUnique<ScopedAllocatorOptimizer>( in InitializeOptimizers()
|