| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_scatter_ops.cc | 25 class GatherOp : public XlaOpKernel { class
27 explicit GatherOp(OpKernelConstruction* context) : XlaOpKernel(context) { in GatherOp() function in tensorflow::__anond94b8f570111::GatherOp
53 GatherOp);
|
| D | gather_op.cc | 234 class GatherOp : public XlaOpKernel { class
236 explicit GatherOp(OpKernelConstruction* context) : XlaOpKernel(context) { in GatherOp() function in tensorflow::GatherOp
257 TF_DISALLOW_COPY_AND_ASSIGN(GatherOp);
264 REGISTER_XLA_OP(Name("Gather"), GatherOp);
265 REGISTER_XLA_OP(Name("GatherV2").CompileTimeConstantInput("axis"), GatherOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_gather_to_torch_index_select.cc | 28 struct GatherIsTorchIndexSelect : public OpRewritePattern<GatherOp> {
29 using OpRewritePattern<GatherOp>::OpRewritePattern;
31 LogicalResult matchAndRewrite(GatherOp gather, in matchAndRewrite()
|
| D | optimize_mhlo.cc | 55 class GatherIsSlice : public OpRewritePattern<GatherOp> {
57 LogicalResult matchAndRewrite(GatherOp gather, in matchAndRewrite()
|
| D | hlo_legalize_to_lhlo.cc | 663 HloToLhloOpConverter<mhlo::GatherOp>, in populateHLOToLHLOConversionPattern()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_op.cc | 38 class GatherOp : public OpKernel { class
45 explicit GatherOp(OpKernelConstruction* c) : OpKernel(c) { in GatherOp() function in tensorflow::GatherOp
198 GatherOp<dev##Device, type, index_type>); \
204 GatherOp<dev##Device, type, index_type>)
|
| D | reduction_gpu_kernels.cu.h | 618 struct GatherOp {
619 __host__ __device__ GatherOp(const int& extent_x, const int& extent_y,
959 GatherOp gather_op(extent_x, extent_y, extent_z, false);
960 typedef gpuprim::TransformInputIterator<int, GatherOp,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_hlo_to_lhlo_op.h | 60 MAP_HLO_TO_LHLO(GatherOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.h | 120 xla::StatusOr<lmhlo::GatherOp> EmitGatherOp(const xla::HloInstruction* instr);
|
| D | mhlo_to_lhlo_with_xla.cc | 1240 xla::StatusOr<lmhlo::GatherOp> LhloDialectEmitter::EmitGatherOp( in EmitGatherOp()
1243 CreateOpWithoutAttrs<lmhlo::GatherOp>(instr)); in EmitGatherOp()
|
| D | legalize_tf.cc | 1655 Value gather = rewriter.create<mhlo::GatherOp>( in matchAndRewrite()
5134 rewriter.replaceOpWithNewOp<mhlo::GatherOp>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 448 } else if (isa<mlir::mhlo::GatherOp, mlir::lmhlo::GatherOp>(op)) { in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 362 MakeAndReturn(GatherOp); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 719 rewriter.replaceOpWithNewOp<TF::GatherOp>(op, op.getType(), input, index, in matchAndRewrite()
738 rewriter.replaceOpWithNewOp<TF::GatherOp>( in matchAndRewrite()
|
| D | prepare_tf.cc | 1333 mhlo::GatherOp::getCanonicalizationPatterns(patterns, context); in ConvertTf2XlaOps()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 2349 static LogicalResult verify(GatherOp op) { in verify()
2370 class GatherFolder final : public OpRewritePattern<GatherOp> {
2372 using OpRewritePattern<GatherOp>::OpRewritePattern;
2373 LogicalResult matchAndRewrite(GatherOp gather, in matchAndRewrite()
2389 void GatherOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1008 class ConvertGatherOp : public OpConversionPattern<mhlo::GatherOp> {
1013 mhlo::GatherOp gather_op, ArrayRef<Value> args, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 198 struct GatherSlice : public OpRewritePattern<GatherOp> {
199 using OpRewritePattern<GatherOp>::OpRewritePattern;
201 LogicalResult matchAndRewrite(GatherOp gather, in matchAndRewrite()
261 void GatherOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1723 auto tf_gather_op = cast<TF::GatherOp>(op); in matchAndRewrite()
1733 rewriter.replaceOpWithNewOp<tosa::GatherOp>( in matchAndRewrite()
1763 rewriter.replaceOpWithNewOp<tosa::GatherOp>( in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/VectorToLLVM/ |
| D | ConvertVectorToLLVM.cpp | 411 : ConvertToLLVMPattern(vector::GatherOp::getOperationName(), context, in VectorGatherOpConversion()
418 auto gather = cast<vector::GatherOp>(op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 293 auto op = builder_.create<mlir::mhlo::GatherOp>( in GatherInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 973 return TFL::GatherOp::build( in BuildGatherOp()
1018 TFL::GatherOp::build( in BuildGatherOp()
|