Home
last modified time | relevance | path

Searched refs:auto_parallel (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Drewriter_config.pb.h989 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() function
2204 ::tensorflow::AutoParallelOptions* auto_parallel) { in unsafe_arena_set_allocated_auto_parallel() argument
2208 _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() argument
2251 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 …]
Drewriter_config.pb.cc1099 static const ::tensorflow::AutoParallelOptions& auto_parallel(const RewriterConfig* msg);
1106 RewriterConfig::_Internal::auto_parallel(const RewriterConfig* msg) { in auto_parallel() function in tensorflow::RewriterConfig::_Internal
1737 InternalWriteMessage(5, _Internal::auto_parallel(this), in _InternalSerialize()
1738 _Internal::auto_parallel(this).GetCachedSize(), target, stream); in _InternalSerialize()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer.cc248 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()
H A DBUILD71 name = "auto_parallel",
72 srcs = ["auto_parallel.cc"],
74 "auto_parallel.h",
94 ":auto_parallel",
627 ":auto_parallel",
/aosp_15_r20/external/tensorflow/tensorflow/security/advisory/
H A Dtfsa-2021-168.md7 …b122e50b4d44ceaaed5a663d635e5c22df/tensorflow/core/grappler/optimizers/auto_parallel.cc#L155-L164):
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/grappler/
H A Dgrappler.h118 TF_TriState auto_parallel; member
H A Dgrappler.cc134 CONFIG_TOGGLE(auto_parallel); in CGraphOptimizerRegister()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Drewriter_config.proto201 AutoParallelOptions auto_parallel = 5; field
215 // (the auto_parallel field) apply to manually requested optimization passes
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dutils.py34 rewriter_config.auto_parallel.enable = False