Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dkernels.cc94 : public ::tensorflow::kernel_factory::OpKernelFactory {
116 tensorflow::kernel_factory::OpKernelRegistrar( in TF_RegisterKernelBuilder()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h257 std::vector<std::unique_ptr<kernel_factory::OpKernelRegistrar>>
Dxla_device.cc536 new kernel_factory::OpKernelRegistrar(def, "XlaCompileOnDemandOp", in RegisterXlaDeviceKernels()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.h262 std::vector<std::unique_ptr<kernel_factory::OpKernelRegistrar>>
Dxla_op_registry.cc316 new kernel_factory::OpKernelRegistrar( in RegisterCompilationKernels()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h1384 static ::tensorflow::kernel_factory::OpKernelRegistrar \
1406 static ::tensorflow::kernel_factory::OpKernelRegistrar \
1438 namespace kernel_factory {
Dop_kernel.cc943 std::unique_ptr<kernel_factory::OpKernelFactory> f) in KernelRegistration()
948 std::unique_ptr<kernel_factory::OpKernelFactory> factory;
1066 namespace kernel_factory { namespace