Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for RegisterOpHandler (0.2 sec)

  2. github.com/tensorflow/tensorflow

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

      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());
  return std::move(corert.get());
}

tfrt::RCArray<tfrt::AsyncValue> CreateTestArguments(const tfrt::Function* func,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 08 08:08:48 UTC 2023
    - 6.3K bytes
    - Viewed (0)
Back to top