- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for perm_attr (0.13 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/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)