Home
last modified time | relevance | path

Searched refs:RetvalOp (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dretval_op.cc30 class RetvalOp : public XlaOpKernel { class
32 explicit RetvalOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in RetvalOp() function in tensorflow::__anon00382dab0111::RetvalOp
63 TF_DISALLOW_COPY_AND_ASSIGN(RetvalOp);
68 RetvalOp);
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc55 RetvalOp::RetvalOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in RetvalOp() function in tensorflow::RetvalOp
60 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 …]
Dfunction_ops.h44 class RetvalOp : public OpKernel {
46 explicit RetvalOp(OpKernelConstruction* ctx);
56 TF_DISALLOW_COPY_AND_ASSIGN(RetvalOp);
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h221 Name(kRetOp).Device(DEVICE).TypeConstraint("T", TYPES), RetvalOp); \
226 RetvalOp); \
228 Name(kDeviceRetOp).Device(DEVICE).TypeConstraint<int32>("T"), RetvalOp); \
282 RetvalOp); \