Searched refs:ReverseOp (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_op.cc | 31 class ReverseOp : public XlaOpKernel { class
33 explicit ReverseOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReverseOp() function in tensorflow::__anon110cc4170111::ReverseOp
67 REGISTER_XLA_OP(Name("Reverse").CompileTimeConstantInput("dims"), ReverseOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_op.cc | 152 class ReverseOp : public OpKernel { class
154 explicit ReverseOp(OpKernelConstruction* context) : OpKernel(context) {} in ReverseOp() function in tensorflow::ReverseOp
300 ReverseOp<CPUDevice, T>) \
353 ReverseOp<GPUDevice, T>) \
380 ReverseOp<CPUDevice, int32>);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.h | 107 xla::StatusOr<lmhlo::ReverseOp> EmitReverseOp(
|
| D | mhlo_to_lhlo_with_xla.cc | 1159 xla::StatusOr<lmhlo::ReverseOp> LhloDialectEmitter::EmitReverseOp( in EmitReverseOp()
1162 CreateOpWithoutAttrs<lmhlo::ReverseOp>(instr)); in EmitReverseOp()
|
| D | legalize_tf.cc | 3147 input = rewriter.create<ReverseOp>( in rewriteWithConstantBegin()
3395 grad = rewriter.create<mhlo::ReverseOp>( in matchAndRewrite()
4227 filter = rewriter.create<ReverseOp>(op.getLoc(), filter, in matchAndRewrite()
5390 input = rewriter.create<ReverseOp>( in matchAndRewrite()
5452 result = rewriter.create<ReverseOp>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 467 } else if (isa<mlir::mhlo::ReverseOp, mlir::lmhlo::ReverseOp>(op)) { in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 579 ->create<mlir::mhlo::ReverseOp>( in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1430 ReverseConverter<lmhlo::ReverseOp>, in populateLHLOToLinalgConversionPattern()
1556 ReverseConverter<mhlo::ReverseOp, false>, in populateHLOToLinalgConversionPattern()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 235 auto op = builder_.create<mlir::mhlo::ReverseOp>( in RevInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1962 auto reverse_op = rewriter.create<tosa::ReverseOp>( in matchAndRewrite()
|
| D | legalize_tfl.cc | 2776 auto reverse_op = rewriter.create<tosa::ReverseOp>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1724 OpFoldResult ReverseOp::fold(ArrayRef<Attribute> operands) { in fold()
|