Searched refs:TF_GetRegisteredKernelsForOp (Results 1 – 6 of 6) sorted by relevance
42 buf = c_api.TF_GetRegisteredKernelsForOp(name)
142 TF_Buffer* buf = TF_GetRegisteredKernelsForOp(op_name, status); in TEST()611 TF_Buffer* buf = TF_GetRegisteredKernelsForOp(op_name, status); in TEST()655 TF_Buffer* buf = TF_GetRegisteredKernelsForOp(op_name, status); in TEST()
1524 TF_CAPI_EXPORT extern TF_Buffer* TF_GetRegisteredKernelsForOp(
2524 TF_GetRegisteredKernelsForOp("TestOpWithSingleKernel", status); in TEST()2537 TF_Buffer* kernel_list_buf = TF_GetRegisteredKernelsForOp("Unknown", status); in TEST()
2480 TF_Buffer* TF_GetRegisteredKernelsForOp(const char* name, TF_Status* status) { in TF_GetRegisteredKernelsForOp() function
948 auto output = TF_GetRegisteredKernelsForOp(name, status.get()); in PYBIND11_MODULE()