Searched refs:CustomKernelCreator (Results 1 – 5 of 5) sorted by relevance
40 void RegisterDefaultCustomKernelCreator(CustomKernelCreator cb);59 CustomKernelCreator custom_kernel_creator,
43 CustomKernelCreator custom_kernel_creator,140 CustomKernelCreator custom_kernel_creator,
148 CustomKernelCreator custom_kernel_creator,198 const CustomKernelCreator custom_kernel_creator_;249 CustomKernelCreator custom_kernel_creator, in FunctionLibraryRuntimeImpl()920 CustomKernelCreator custom_creator = nullptr;922 void Set(CustomKernelCreator cb) { in Set()927 CustomKernelCreator Get() { in Get()940 void RegisterDefaultCustomKernelCreator(CustomKernelCreator cb) { in RegisterDefaultCustomKernelCreator()948 CustomKernelCreator custom_kernel_creator, in NewFunctionLibraryRuntime()
54 CustomKernelCreator custom_kernel_creator, in ProcessFunctionLibraryRuntime()338 CustomKernelCreator custom_kernel_creator, in Clone()
562 CustomKernelCreator; typedef