- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for perm_attr (0.09 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
TensorType perm_type = operand_type.cloneWith( {static_cast<int64_t>(permutation.size())}, rewriter.getI32Type()); auto perm_attr = DenseIntElementsAttr::get(perm_type, permutation); auto perm = rewriter.create<arith::ConstantOp>(op.getLoc(), perm_attr); Value input = op.getOperand(); return rewriter.create<TFL::TransposeOp>(op.getLoc(), input, perm); }
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/lite/transforms/optimize.cc
const int rank = perm.getNumElements(); auto perm_iter = perm.getValues<APInt>(); if (rank < 3) return false; for (int i = 2; i < rank; i++) { if (perm_iter[i] != i - 2) { return false; } } return perm_iter[0] == rank - 2 && perm_iter[1] == rank - 1; } }; struct FuseLogSoftmax : public OpRewritePattern<TFL::SubOp> {
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/tensorflow/ir/tf_ops_n_z.cc
Attribute zero_attr; if (element_type.isIntOrFloat()) zero_attr = rewriter.getZeroAttr(type); else if (element_type.isa<TF::StringType>()) zero_attr = DenseStringElementsAttr::get(type, {""}); if (!zero_attr) return failure(); auto zero_const = rewriter.create<TF::ConstOp>(op.getLoc(), zero_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)