Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Dcluster_function_library_runtime.cc47 void EagerClusterFunctionLibraryRuntime::Instantiate( in Instantiate()
114 void EagerClusterFunctionLibraryRuntime::Run( in Run()
143 void EagerClusterFunctionLibraryRuntime::Run( in Run()
256 void EagerClusterFunctionLibraryRuntime::CleanUp( in CleanUp()
288 return new EagerClusterFunctionLibraryRuntime( in CreateClusterFLR()
H A Dcluster_function_library_runtime.h37 class EagerClusterFunctionLibraryRuntime
40 EagerClusterFunctionLibraryRuntime(const uint64 context_id, EagerContext* ctx, in EagerClusterFunctionLibraryRuntime() function
46 ~EagerClusterFunctionLibraryRuntime() override{}; in ~EagerClusterFunctionLibraryRuntime()
H A Deager_service_impl_test.cc780 eager_cluster_flr_ = std::make_unique<EagerClusterFunctionLibraryRuntime>( in Init()
836 std::unique_ptr<EagerClusterFunctionLibraryRuntime> eager_cluster_flr_;
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext_distributed_manager.cc757 eager::EagerClusterFunctionLibraryRuntime* cluster_flr = in EnableCollectiveOps()
758 new eager::EagerClusterFunctionLibraryRuntime( in EnableCollectiveOps()