Searched refs:custom_kernel_creator_ (Results 1 – 1 of 1) sorted by relevance
368 const CustomKernelCreator custom_kernel_creator_; member in tensorflow::FunctionLibraryRuntimeImpl437 custom_kernel_creator_(std::move(custom_kernel_creator)), in FunctionLibraryRuntimeImpl()542 if (custom_kernel_creator_) { in CreateKernel()544 s = custom_kernel_creator_(this, ndef, &ret); in CreateKernel()1182 custom_kernel_creator_, out_lib_def, out_pflr)); in Clone()