- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for runtime_fallback_executor (0.17 sec)
-
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.cc
int CurrentThreadId() const override { return tpool_.CurrentThreadId(); } void Cancel() override {} private: ThreadPool tpool_; }; } // namespace RuntimeFallbackExecutor::RuntimeFallbackExecutor(int64_t num_threads) : intra_op_(std::make_unique<IntraOpThreadPool>(num_threads)) { // Create a HostContext for running TFRT functions. Concurrent work queue acts
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.h
#include "tfrt/host_context/resource_context.h" // from @tf_runtime #include "tfrt/support/ref_count.h" // from @tf_runtime namespace tensorflow { class RuntimeFallbackExecutor { public: explicit RuntimeFallbackExecutor(int64_t num_threads); // Prepare() needs to be called once before calling Execute(). It sets up all // things necessary to execute the given 'mlir_input' with the fallback to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 14 22:40:40 UTC 2022 - 2.4K bytes - Viewed (0)