Searched refs:auto_parallel (Results 1 – 9 of 9) sorted by relevance
989 const ::tensorflow::AutoParallelOptions& auto_parallel() const; in Swap()992 void set_allocated_auto_parallel(::tensorflow::AutoParallelOptions* auto_parallel); in Swap()998 ::tensorflow::AutoParallelOptions* auto_parallel); in Swap()2199 inline const ::tensorflow::AutoParallelOptions& RewriterConfig::auto_parallel() const { in auto_parallel() function2204 ::tensorflow::AutoParallelOptions* auto_parallel) { in unsafe_arena_set_allocated_auto_parallel() argument2208 _impl_.auto_parallel_ = auto_parallel; in unsafe_arena_set_allocated_auto_parallel()2246 …oid RewriterConfig::set_allocated_auto_parallel(::tensorflow::AutoParallelOptions* auto_parallel) { in set_allocated_auto_parallel() argument2251 if (auto_parallel) { in set_allocated_auto_parallel()2253 ::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(auto_parallel); in set_allocated_auto_parallel()2255 auto_parallel = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage( in set_allocated_auto_parallel()[all …]
1099 static const ::tensorflow::AutoParallelOptions& auto_parallel(const RewriterConfig* msg);1106 RewriterConfig::_Internal::auto_parallel(const RewriterConfig* msg) { in auto_parallel() function in tensorflow::RewriterConfig::_Internal1737 InternalWriteMessage(5, _Internal::auto_parallel(this), in _InternalSerialize()1738 _Internal::auto_parallel(this).GetCachedSize(), target, stream); in _InternalSerialize()
248 new AutoParallel(cfg_.auto_parallel().num_replicas())); in MakeNewOptimizer()476 if (cfg_.auto_parallel().enable() && PLUGIN_IS_ON(auto_parallel)) { in InitializeOptimizers()478 MakeUnique<AutoParallel>(cfg_.auto_parallel().num_replicas())); in InitializeOptimizers()657 user_cfg.toggle_config["auto_parallel"] = cfg_.auto_parallel().enable() in PrintUserAndPluginConfigs()1341 rewrite_cfg.auto_parallel().enable() || in MetaOptimizerEnabled()
71 name = "auto_parallel",72 srcs = ["auto_parallel.cc"],74 "auto_parallel.h",94 ":auto_parallel",627 ":auto_parallel",
7 …b122e50b4d44ceaaed5a663d635e5c22df/tensorflow/core/grappler/optimizers/auto_parallel.cc#L155-L164):
118 TF_TriState auto_parallel; member
134 CONFIG_TOGGLE(auto_parallel); in CGraphOptimizerRegister()
201 AutoParallelOptions auto_parallel = 5; field215 // (the auto_parallel field) apply to manually requested optimization passes
34 rewriter_config.auto_parallel.enable = False