- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for perm_attr (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
} } // Nothing to do here. if (!permutation_op || transpose_ops.empty()) return; SmallVector<int64_t, 4> permutation; auto perm_attr = mlir::cast<DenseElementsAttr>(permutation_op.getValue()); for (const auto& value : perm_attr.getValues<APInt>()) permutation.push_back(value.getSExtValue()); // We want to make sure the shape of the operand equals the transposed shape.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
auto perm_type = tensorflow::GetTypeFromTFTensorShape( {static_cast<int>(perm.size())}, rewriter.getIntegerType(32)); auto perm_attr = DenseElementsAttr::get(perm_type, llvm::ArrayRef<int>(perm)); auto perm_op = rewriter.create<TF::ConstOp>(loc, perm_type, perm_attr); // Create tensor type for the transpose result. auto filter_type = mlir::cast<RankedTensorType>(filter.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
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/tf2xla/internal/passes/xla_broadcast.cc
return true; } // Create a dummy zero to be fed locally from the host to the TPUExecute. Value CreateZeroInput(Location loc, OpBuilder& builder, Attribute zero_attr, DenseIntElementsAttr shape_attr) { ConstOp zero = builder.create<ConstOp>(loc, zero_attr); zero->setAttr(kICIWeightDistributionMlirBridgeMarker, builder.getBoolAttr(true)); ConstOp shape = builder.create<ConstOp>(loc, shape_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
} auto perm_value = llvm::dyn_cast<DenseElementsAttr>(perm.getValue()).getValues<int32_t>(); llvm::SmallVector<int32_t> perm_arr(perm_value.begin(), perm_value.end()); // Compute inverse of input transpose. llvm::SmallVector<int32_t> inverse_perm = InvertPermutation(perm_arr); if (tpose_arg.getResult().getType() != op->getResult(0).getType()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
src/test/java/org/codelibs/opensearch/extension/analysis/NGramSynonymTokenizer.java
private int finalOffset; private final PriorityQueue<MyToken> queue; private MyToken prevToken; private final List<MyToken> synonyms; private final CharTermAttribute termAttr = addAttribute(CharTermAttribute.class); private final OffsetAttribute offsetAttr = addAttribute(OffsetAttribute.class);
Registered: Wed Jun 12 15:38:08 UTC 2024 - Last Modified: Thu Feb 22 01:36:54 UTC 2024 - 17K 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) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// as specified by element_dtype. RankedTensorType zero_type = tensorflow::GetTypeFromTFTensorShape({}, element_dtype); auto zero_attr = rewriter.getZeroAttr(zero_type); auto zero = rewriter.create<arith::ConstantOp>(loc, zero_type, zero_attr); rewriter.replaceOpWithNewOp<TF::FillOp>(op, result_type, list_shape, zero); return success(); } }; struct ConvertTensorListReserve
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// Note on zero_attr integer type: DynamicUpdateSlice op start_indices are // required to have matching types. This rewrite rule creates // DynamicUpdateSlice ops where the first "start index" is always i32 and // subsequent ones are constructed based on zero_attr. Thus the type // for zero_attr needs to be i32 as well. auto zero_attr = IntegerAttr::get(rewriter.getIntegerType(32), 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)