Home
last modified time | relevance | path

Searched refs:ReverseV2Op (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreverse_op.cc229 class ReverseV2Op : public OpKernel { class
231 explicit ReverseV2Op(OpKernelConstruction* context) : OpKernel(context) {} in ReverseV2Op() function in tensorflow::ReverseV2Op
306 ReverseV2Op<CPUDevice, T, int32>) \
312 ReverseV2Op<CPUDevice, T, int64>)
359 ReverseV2Op<GPUDevice, T, int32>) \
365 ReverseV2Op<GPUDevice, T, int64>)
388 ReverseV2Op<CPUDevice, int32, int32>);
396 ReverseV2Op<CPUDevice, int32, int64>);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreverse_op.cc69 class ReverseV2Op : public XlaOpKernel { class
71 explicit ReverseV2Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReverseV2Op() function in tensorflow::__anon110cc4170111::ReverseV2Op
120 ReverseV2Op);
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc120 return builder->create<TF::ReverseV2Op>(location, type, value_to_reverse, in Reverse()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1940 auto tf_reverse_op = cast<TF::ReverseV2Op>(op); in matchAndRewrite()
Dlegalize_tfl.cc2753 auto tfl_reverse_op = cast<TFL::ReverseV2Op>(op); in matchAndRewrite()