Home
last modified time | relevance | path

Searched refs:OpKernelRegistrar (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h1426 ::tensorflow::kernel_factory::OpKernelRegistrar registrar( \
1505 class OpKernelRegistrar {
1510 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() function
1518 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() function
Dop_kernel.cc1321 void OpKernelRegistrar::InitInternal(const KernelDef* kernel_def, in InitInternal()
1344 OpKernel* OpKernelRegistrar::PtrOpKernelFactory::Create( in Create()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h300 std::vector<std::unique_ptr<kernel_factory::OpKernelRegistrar>>
Dxla_device.cc610 new kernel_factory::OpKernelRegistrar(def, "XlaCompileOnDemandOp", in RegisterXlaDeviceKernels()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.h319 std::vector<std::unique_ptr<kernel_factory::OpKernelRegistrar>>
Dxla_op_registry.cc331 new kernel_factory::OpKernelRegistrar( in RegisterCompilationKernels()
/external/tensorflow/tensorflow/c/
Dkernels.cc211 tensorflow::kernel_factory::OpKernelRegistrar( in TF_RegisterKernelBuilderWithKernelDef()
227 tensorflow::kernel_factory::OpKernelRegistrar( in TF_RegisterKernelBuilderWithKernelDef()