- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for tfd (0.02 sec)
-
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) -
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)