Searched refs:PyFuncOp (Results 1 – 1 of 1) sorted by relevance
484 class PyFuncOp : public OpKernel { class486 explicit PyFuncOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in PyFuncOp() function in tensorflow::PyFuncOp564 TF_DISALLOW_COPY_AND_ASSIGN(PyFuncOp);567 REGISTER_KERNEL_BUILDER(Name("PyFunc").Device(DEVICE_CPU), PyFuncOp);568 REGISTER_KERNEL_BUILDER(Name("PyFuncStateless").Device(DEVICE_CPU), PyFuncOp);569 REGISTER_KERNEL_BUILDER(Name("EagerPyFunc").Device(DEVICE_CPU), PyFuncOp);570 REGISTER_KERNEL_BUILDER(Name("EagerPyFunc").Device(DEVICE_GPU), PyFuncOp);