- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for permutation (0.29 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
if (!isKernelFormatHWOI(dnums)) { SmallVector<int64_t, 4> permutation; for (int64_t dim : dnums.getKernelSpatialDimensions()) { permutation.push_back(dim); } permutation.push_back(dnums.getKernelOutputFeatureDimension()); permutation.push_back(dnums.getKernelInputFeatureDimension()); auto filter_transposed = rewriter.create<mhlo::TransposeOp>(
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/tensorflow/ir/tf_ops_a_m.cc
template <class Op> static LogicalResult InferenceFoldOperandsPermutation( ArrayRef<int64_t> permutation, Op* op) { // FusedBatchNorm in training mode is a layout sentitive operation, and should // have already assigned an optimal data format. if (op->getIsTraining()) return failure(); return ::mlir::TF::FoldOperandsPermutation(permutation, op); } template <class Op>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
operand_type.cloneWith(shape, rewriter.getI32Type()); // Cast permutation attribute from i64 to i32 as they are required to be i32 // in TFLite. SmallVector<int32_t> permutation_i32 = CastI64ArrayToI32(op.getPermutation()).value(); auto permutation_attr = DenseIntElementsAttr::get(permutation_type, permutation_i32); auto permutation =
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
LogicalResult PadOp::FoldOperandsPermutation(ArrayRef<int64_t> permutation) { // Paddings must be defined by a constant operation. auto paddings_op = dyn_cast_or_null<TF::ConstOp>(getPaddings().getDefiningOp()); if (!paddings_op) return failure(); auto paddings_value = paddings_op.getValue().dyn_cast<DenseElementsAttr>(); if (!paddings_value || paddings_value.getNumElements() != permutation.size() * 2) return failure();
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/transforms/optimize.cc
if (a.getRank() != 2 || b.getRank() < 2) return false; if (a.getShape().back() != b.getShape().back()) return false; return b.getDimSize(b.getRank() - 2) == 1; } // Checks if the transpose permutation has value [2, 3, ..., n-1, 0, 1]. bool TransposeFirstTwoDimToLast(DenseIntElementsAttr perm) const { int rank = perm.getNumElements(); if (rank < 3) return false; for (int i = 0; i < rank - 2; i++) {
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/ir/tfl_ops.cc
llvm::ArrayRef(return_squeeze_shape)); } // This is a utility function to deduce the effective permutation to apply on // TFL_TransposeOp when the tensor has some dimensions with value==1 // Example- "tfl.transpose"(tensor<56x8x56x1x1x1x7xf32>, [4, 5, 1, 2, 0, 6, 3]) // Permutation before squeese is [4, 5, 1, 2, 0, 6, 3] becomes [1, 2, 0, 3] // after squeeze is perfomed to retain the relative ordering of the non-1 dims.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
src/cmd/compile/internal/ssa/_gen/generic.rules
// blocks are specified using the following format: // (kind controlvalue succ0 succ1 ...) // controlvalue must be "nil" or a value expression // succ* fields must be variables // For now, the generated successors must be a permutation of the matched successors. // constant folding (Trunc16to8 (Const16 [c])) => (Const8 [int8(c)]) (Trunc32to8 (Const32 [c])) => (Const8 [int8(c)]) (Trunc32to16 (Const32 [c])) => (Const16 [int16(c)])
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu May 16 22:21:05 UTC 2024 - 135.3K bytes - Viewed (0)