Searched refs:CpuCastOp (Results 1 – 2 of 2) sorted by relevance
120 CpuCastOp::CpuCastOp(OpKernelConstruction* ctx) : CastOpBase(ctx) { in CpuCastOp() function in tensorflow::CpuCastOp124 Status CpuCastOp::Prepare() { in Prepare()222 REGISTER_KERNEL_BUILDER(Name("Cast").Device(DEVICE_CPU), CpuCastOp);282 REGISTER_KERNEL_BUILDER(Name("_HostCast").Device(DEVICE_CPU), CpuCastOp);285 CpuCastOp);
112 class CpuCastOp : public CastOpBase {114 explicit CpuCastOp(OpKernelConstruction* ctx);