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::__anon00382dab0111::RetvalOp63 TF_DISALLOW_COPY_AND_ASSIGN(RetvalOp);68 RetvalOp);
55 RetvalOp::RetvalOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in RetvalOp() function in tensorflow::RetvalOp60 void RetvalOp::Compute(OpKernelContext* ctx) { in Compute()73 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kRetOp).Device(DEVICE_CPU), RetvalOp);74 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceRetOp).Device(DEVICE_CPU), RetvalOp);89 Name(kRetOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), RetvalOp);95 RetvalOp);130 Name(kRetOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), RetvalOp);138 RetvalOp);140 Name(kDeviceRetOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), RetvalOp);146 RetvalOp);[all …]
44 class RetvalOp : public OpKernel {46 explicit RetvalOp(OpKernelConstruction* ctx);56 TF_DISALLOW_COPY_AND_ASSIGN(RetvalOp);
221 Name(kRetOp).Device(DEVICE).TypeConstraint("T", TYPES), RetvalOp); \226 RetvalOp); \228 Name(kDeviceRetOp).Device(DEVICE).TypeConstraint<int32>("T"), RetvalOp); \282 RetvalOp); \