Home
last modified time | relevance | path

Searched refs:ReverseOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreverse_op.cc31 class ReverseOp : public XlaOpKernel { class
33 explicit ReverseOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReverseOp() function in tensorflow::__anona7f017790111::ReverseOp
67 REGISTER_XLA_OP(Name("Reverse").CompileTimeConstantInput("dims"), ReverseOp);
/external/tensorflow/tensorflow/core/kernels/
Dreverse_op.cc155 class ReverseOp : public OpKernel { class
157 explicit ReverseOp(OpKernelConstruction* context) : OpKernel(context) {} in ReverseOp() function in tensorflow::ReverseOp
303 ReverseOp<CPUDevice, T>) \
361 ReverseOp<GPUDevice, T>) \
393 ReverseOp<CPUDevice, int32>);
418 ReverseOp<SYCLDevice, T>) \
442 ReverseOp<CPUDevice, int32>);