Home
last modified time | relevance | path

Searched refs:RegisterDefaultCustomKernelCreator (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.h41 void RegisterDefaultCustomKernelCreator(CustomKernelCreator cb);
Dfunction.cc1215 void RegisterDefaultCustomKernelCreator(CustomKernelCreator cb) { in RegisterDefaultCustomKernelCreator() function
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op.cc245 RegisterDefaultCustomKernelCreator(CreateXlaLaunchOp); in RegisterLaunchOpCreator()