Searched refs:custom_kernel_creator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 46 CustomKernelCreator custom_kernel_creator,
218 CustomKernelCreator custom_kernel_creator,
|
| D | process_function_library_runtime.cc | 87 CustomKernelCreator custom_kernel_creator, in ProcessFunctionLibraryRuntime() argument
99 optimizer_options, std::move(custom_kernel_creator), this); in ProcessFunctionLibraryRuntime()
105 optimizer_options, custom_kernel_creator, this); in ProcessFunctionLibraryRuntime()
1005 CustomKernelCreator custom_kernel_creator, in Clone() argument
1011 optimizer_options, std::move(custom_kernel_creator), default_thread_pool_, in Clone()
|
| D | function.h | 60 CustomKernelCreator custom_kernel_creator,
|
| D | function.cc | 312 CustomKernelCreator custom_kernel_creator,
429 CustomKernelCreator custom_kernel_creator, in FunctionLibraryRuntimeImpl() argument
437 custom_kernel_creator_(std::move(custom_kernel_creator)), in FunctionLibraryRuntimeImpl()
1223 CustomKernelCreator custom_kernel_creator, in NewFunctionLibraryRuntime() argument
1227 optimizer_options, std::move(custom_kernel_creator), parent)); in NewFunctionLibraryRuntime()
|