Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for tfd (0.02 sec)

  1. tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/batch_function_fallback_benchmark_test.cc

                                  diag) { LOG(ERROR) << diag.message(); },
          tfrt::CreateMallocAllocator(),
          tfrt::CreateMultiThreadedWorkQueue(16, 16));
      CHECK(corert);
      auto fallback_op_handler = tensorflow::tfd::CreateKernelFallbackOpHandler(
          corert->get(), corert->get()->GetHostContext()->GetHostDeviceRef());
      CHECK(fallback_op_handler);
      corert.get()->RegisterOpHandler("tfkernel", fallback_op_handler.get());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 08 08:08:48 UTC 2023
    - 6.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.cc

          resource_context_
              .GetOrCreateResource<tensorflow::tfd::EagerContextResource>(
                  tensorflow::tfd::kEagerContextResourceName);
      auto expected_eager_context = eager_context_resource->GetTFEagerContext();
      auto* eager_context = expected_eager_context.get();
    
      // Initialize fallback kernels state with a custom intra-op thread pool.
      auto status = tensorflow::tfd::SetUpKernelFallbackCompatRequestContext(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 00:18:59 UTC 2024
    - 9.1K bytes
    - Viewed (0)
Back to top