Searched refs:_HostConstantOp (Results 1 – 3 of 3) sorted by relevance
27 _HostConstantOp::_HostConstantOp(OpKernelConstruction* ctx) in _HostConstantOp() function in tensorflow::_HostConstantOp42 void _HostConstantOp::Compute(OpKernelContext* ctx) { in Compute()54 _HostConstantOp);62 _HostConstantOp);66 REGISTER_KERNEL_BUILDER(Name("HostConst").Device(DEVICE_CPU), _HostConstantOp);68 Name("HostConst").Device(DEVICE_GPU).HostMemory("output"), _HostConstantOp);72 _HostConstantOp);
28 class _HostConstantOp : public OpKernel {30 explicit _HostConstantOp(OpKernelConstruction* ctx);33 ~_HostConstantOp() override {} in ~_HostConstantOp()37 TF_DISALLOW_COPY_AND_ASSIGN(_HostConstantOp);
98 Name("HostConst").Device(DEVICE).HostMemory("output"), _HostConstantOp); \