Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for runtime_fallback_executor (0.18 sec)

  1. 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)
  2. 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)
  3. tensorflow/compiler/mlir/tfrt/BUILD

            "@tf_runtime//:tensor_opdefs",
        ],
    )
    
    cc_library(
        name = "runtime_fallback_executor",
        testonly = True,
        srcs = [
            "runtime_fallback/runtime_fallback_executor.cc",
        ],
        hdrs = [
            "runtime_fallback/runtime_fallback_executor.h",
        ],
        deps = [
            ":host_context_util",
            ":tf_to_tfrt",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 19:04:21 UTC 2024
    - 23.6K bytes
    - Viewed (0)
Back to top