Searched refs:ScopedAllocatorOptimizer (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.h | 34 class ScopedAllocatorOptimizer; variable
38 class ScopedAllocatorOptimizer : public GraphOptimizer {
40 ScopedAllocatorOptimizer(RewriterConfig::Toggle opt_level,
42 ~ScopedAllocatorOptimizer() override;
93 virtual Status Rewrite(ScopedAllocatorOptimizer* paopti,
|
| D | scoped_allocator_optimizer.cc | 215 Status MaybeRewriteInput(ScopedAllocatorOptimizer* sa_opti, in MaybeRewriteInput()
258 Status GetInputs(ScopedAllocatorOptimizer* sa_opti, int64 invocation_count, in GetInputs()
347 void ScopedAllocatorOptimizer::ExtendNodeAttr(StringPiece name, in ExtendNodeAttr()
362 class UnaryElementwiseRewriter : public ScopedAllocatorOptimizer::Rewriter {
446 Status AnalyzeInputs(ScopedAllocatorOptimizer* sa_opti, in AnalyzeInputs()
511 ScopedAllocatorOptimizer* sa_opti, GraphDef* graph, NodeMap* node_map, in ConstructScopedAllocatorNode()
553 ScopedAllocatorOptimizer::ExtendNodeAttr(kScopedAllocatorAttrName, in ConstructScopedAllocatorNode()
821 Status Rewrite(ScopedAllocatorOptimizer* sa_opti, int64 invocation_count, in Rewrite()
889 ScopedAllocatorOptimizer::ScopedAllocatorOptimizer( in ScopedAllocatorOptimizer() function in tensorflow::grappler::ScopedAllocatorOptimizer
909 Status ScopedAllocatorOptimizer::Optimize(Cluster* /*cluster*/, in Optimize()
[all …]
|
| D | scoped_allocator_optimizer_test.cc | 369 ScopedAllocatorOptimizer sao(RewriterConfig::ON, opts); in TEST_F()
370 ScopedAllocatorOptimizer::OpNameSet ons; in TEST_F()
447 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {0, 2}, &nd); in TEST_F()
448 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {6, 7}, &nd); in TEST_F()
449 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {2, 3}, &nd); in TEST_F()
488 ScopedAllocatorOptimizer sao(RewriterConfig::ON, opts); in TEST_F()
489 ScopedAllocatorOptimizer::OpNameSet ons; in TEST_F()
520 ScopedAllocatorOptimizer sao(RewriterConfig::ON, opts); in TEST_F()
521 ScopedAllocatorOptimizer::OpNameSet ons; in TEST_F()
572 ScopedAllocatorOptimizer sao(RewriterConfig::ON, opts); in TEST_F()
[all …]
|
| D | meta_optimizer.cc | 212 new ScopedAllocatorOptimizer(cfg_.scoped_allocator_optimization(), in MakeNewOptimizer()
312 optimizers->push_back(MakeUnique<ScopedAllocatorOptimizer>( in InitializeOptimizers()
|