Searched refs:RetvalOp (Results 1 – 4 of 4) sorted by relevance
30 class RetvalOp : public XlaOpKernel { class32 explicit RetvalOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in RetvalOp() function in tensorflow::__anon8a03a7020111::RetvalOp65 TF_DISALLOW_COPY_AND_ASSIGN(RetvalOp);70 RetvalOp);
71 RetvalOp::RetvalOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in RetvalOp() function in tensorflow::RetvalOp76 void RetvalOp::Compute(OpKernelContext* ctx) { in Compute()89 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kRetOp).Device(DEVICE_CPU), RetvalOp);90 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceRetOp).Device(DEVICE_CPU), RetvalOp);95 REGISTER_KERNEL_BUILDER(Name(kRetOp).Device(DEVICE_TPU_SYSTEM), RetvalOp);129 Name(kRetOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), RetvalOp);137 RetvalOp);139 Name(kDeviceRetOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), RetvalOp);145 RetvalOp);151 RetvalOp);
44 class RetvalOp : public OpKernel {46 explicit RetvalOp(OpKernelConstruction* ctx);56 TF_DISALLOW_COPY_AND_ASSIGN(RetvalOp);
166 Name(kRetOp).Device(DEVICE).TypeConstraint("T", TYPES), RetvalOp); \171 RetvalOp); \173 Name(kDeviceRetOp).Device(DEVICE).TypeConstraint<int32>("T"), RetvalOp); \231 RetvalOp);