Searched refs:CustomKernelCreator (Results 1 – 5 of 5) sorted by relevance
41 void RegisterDefaultCustomKernelCreator(CustomKernelCreator cb);60 CustomKernelCreator custom_kernel_creator,
46 CustomKernelCreator custom_kernel_creator,218 CustomKernelCreator custom_kernel_creator,
312 CustomKernelCreator custom_kernel_creator,368 const CustomKernelCreator custom_kernel_creator_;429 CustomKernelCreator custom_kernel_creator, in FunctionLibraryRuntimeImpl()1195 CustomKernelCreator custom_creator = nullptr;1197 void Set(CustomKernelCreator cb) { in Set()1202 CustomKernelCreator Get() { in Get()1215 void RegisterDefaultCustomKernelCreator(CustomKernelCreator cb) { in RegisterDefaultCustomKernelCreator()1223 CustomKernelCreator custom_kernel_creator, in NewFunctionLibraryRuntime()
87 CustomKernelCreator custom_kernel_creator, in ProcessFunctionLibraryRuntime()1005 CustomKernelCreator custom_kernel_creator, in Clone()
720 CustomKernelCreator; typedef