Searched defs:CpuCastOp (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dcast_op.cc76 CpuCastOp::CpuCastOp(OpKernelConstruction* ctx) : CastOpBase(ctx) { function in class:tensorflow::CpuCastOp
80 Status CpuCastOp::Prepare() {
169 REGISTER_KERNEL_BUILDER(Name("Cast").Device(DEVICE_CPU), CpuCastOp); member in namespace:tensorflow
245 REGISTER_KERNEL_BUILDER(Name("_HostCast").Device(DEVICE_CPU), CpuCastOp); member in namespace:tensorflow
248 CpuCastOp); member in namespace:tensorflow
252 CpuCastOp); member in namespace:tensorflow
H A Dcast_op.h47 class CpuCastOp : public CastOpBase { class in namespace:tensorflow
49 explicit CpuCastOp(OpKernelConstruction* ctx);

Completed in 178 milliseconds