Searched refs:ReverseV2Op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_op.cc | 232 class ReverseV2Op : public OpKernel { class
234 explicit ReverseV2Op(OpKernelConstruction* context) : OpKernel(context) {} in ReverseV2Op() function in tensorflow::ReverseV2Op
309 ReverseV2Op<CPUDevice, T, int32>) \
315 ReverseV2Op<CPUDevice, T, int64>)
367 ReverseV2Op<GPUDevice, T, int32>) \
373 ReverseV2Op<GPUDevice, T, int64>)
401 ReverseV2Op<CPUDevice, int32, int32>);
409 ReverseV2Op<CPUDevice, int32, int64>);
424 ReverseV2Op<SYCLDevice, T, int32>) \
430 ReverseV2Op<SYCLDevice, T, int64>)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_op.cc | 69 class ReverseV2Op : public XlaOpKernel { class
71 explicit ReverseV2Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReverseV2Op() function in tensorflow::__anona7f017790111::ReverseV2Op
120 ReverseV2Op);
|