- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for transpose_ops (0.25 sec)
-
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) -
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) -
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) -
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) -
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) -
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)