Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for transpose_ops (0.25 sec)

  1. tensorflow/compiler/mlir/lite/transforms/optimize.cc

          return failure();
        }
    
        auto transpose_op = reshape_op_1.getInput().getDefiningOp<TransposeOp>();
        if (!transpose_op) {
          return failure();
        }
        DenseIntElementsAttr transpose_perm_1;
        if (!matchPattern(transpose_op.getPerm(), m_Constant(&transpose_perm_1)) ||
            !TransposeFirstTwoDimToLast(transpose_perm_1)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

    class RewriteQuantizedTransposeOp
        : public OpRewritePattern<stablehlo::TransposeOp> {
     public:
      using OpRewritePattern<stablehlo::TransposeOp>::OpRewritePattern;
    
      LogicalResult match(stablehlo::TransposeOp op) const override {
        return success(IsOpFullyQuantized(op));
      }
    
      void rewrite(stablehlo::TransposeOp op,
                   PatternRewriter& rewriter) const override {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      }
    
      return success();
    }
    
    //===----------------------------------------------------------------------===//
    // TransposeOp
    //===----------------------------------------------------------------------===//
    
    LogicalResult TransposeOp::verify() {
      TransposeOp op = *this;
      auto perm_type = op.getPerm().getType().dyn_cast<RankedTensorType>();
      auto x_type = op.getX().getType().dyn_cast<RankedTensorType>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

                  dnums.getOutputFeatureDimension(),
                  *dnums.getOutputSpatialDimensions().begin(), num_spatial_dims,
                  conv_output_type, rewriter);
          output = rewriter.create<mhlo::TransposeOp>(
              conv_op.getLoc(), conv_op.getType(), output, permutation);
        }
        rewriter.replaceOp(conv_op, {output});
      }
    };
    
    // Convert a 1-D convolution into a 2-D convolution (which TF supports) so that
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

        }
      }
    }
    
    }  // namespace
    
    void TransposeOp::getCanonicalizationPatterns(RewritePatternSet& results,
                                                  MLIRContext* context) {
      results.add<ConvertTransposeToDecreaseRank>(context);
    }
    
    OpFoldResult TransposeOp::fold(FoldAdaptor adaptor) {
      auto operands = adaptor.getOperands();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

    class TFL_OperandHasRankAtMost<int n, int m> :
      PredOpTrait<"operand " # n # " is at most " # m # "-D",
        TFL_OperandHasRankAtMostPred<n, m>>;
    
    // Not all dimentions in the tensor will contribute to the data move in a
    // TransposeOp. Effective rank is the number of dimentions != 1
    class TFL_TransposeOperandHasEffectiveRankAtMost<int n, int m> :
      PredOpTrait<"operand " # n # " is at most " # m # "-D",
        Or<[TFL_OperandIsUnrankedPred<n>,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top