Searched refs:CastOpBase (Results 1 – 3 of 3) sorted by relevance
57 CastOpBase::CastOpBase(OpKernelConstruction* ctx) : OpKernel(ctx) { in CastOpBase() function in tensorflow::CastOpBase95 void CastOpBase::Compute(OpKernelContext* ctx) { in Compute()116 Status CastOpBase::Unimplemented() { in Unimplemented()122 CpuCastOp::CpuCastOp(OpKernelConstruction* ctx) : CastOpBase(ctx) { in CpuCastOp()172 class GpuCastOp : public CastOpBase {174 explicit GpuCastOp(OpKernelConstruction* ctx) : CastOpBase(ctx) { in GpuCastOp()254 class SyclCastOp : public CastOpBase {256 explicit SyclCastOp(OpKernelConstruction* ctx) : CastOpBase(ctx) { in SyclCastOp()
93 class CastOpBase : public OpKernel {95 explicit CastOpBase(OpKernelConstruction* ctx);108 TF_DISALLOW_COPY_AND_ASSIGN(CastOpBase);112 class CpuCastOp : public CastOpBase {
613 friend class CastOpBase; // For access to set_dtype; variable