Searched refs:CastOpBase (Results 1 – 3 of 3) sorted by relevance
54 CastOpBase::CastOpBase(OpKernelConstruction* ctx) : OpKernel(ctx) { in CastOpBase() function in tensorflow::CastOpBase92 void CastOpBase::Compute(OpKernelContext* ctx) { in Compute()114 Status CastOpBase::Unimplemented() { in Unimplemented()120 CpuCastOp::CpuCastOp(OpKernelConstruction* ctx) : CastOpBase(ctx) { in CpuCastOp()171 class GpuCastOp : public CastOpBase {173 explicit GpuCastOp(OpKernelConstruction* ctx) : CastOpBase(ctx) { in GpuCastOp()
93 class CastOpBase : public OpKernel {95 explicit CastOpBase(OpKernelConstruction* ctx);108 TF_DISALLOW_COPY_AND_ASSIGN(CastOpBase);112 class CpuCastOp : public CastOpBase {
679 friend class CastOpBase; // For access to set_dtype. variable