- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for DenseFPElementsAttr (0.16 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize.cc
// operand if necessary. DenseFPElementsAttr cst; if (!matchPattern(binary_op->getOperand(1), m_Constant(&cst))) return failure(); if (cst.getNumElements() != 1) return failure(); APFloat cst_value = *cst.value_begin<APFloat>(); // Affine op. Value filter = fc_op.getFilter(); Value bias = fc_op.getBias(); DenseFPElementsAttr filter_cst, bias_cst;
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/legalize_hlo.cc
&data_format)) { return rewriter.notifyMatchFailure( div_op, "not the root of spatial pooling without dilation"); } DenseFPElementsAttr divisor; auto div_rhs = recursivelyWalkUp<mhlo::BroadcastInDimOp>(div_op.getRhs()); if (matchPattern(div_rhs, m_Constant(&divisor))) {
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/flatbuffer_export.cc
auto layer_stats = mlir::cast<mlir::DenseFPElementsAttr>(stats_op.getLayerStats()); std::optional<mlir::ElementsAttr> axis_stats = stats_op.getAxisStats(); std::optional<uint64_t> axis = stats_op.getAxis(); std::vector<float> mins, maxs; mlir::DenseFPElementsAttr min_max_attr = axis_stats.has_value() ? mlir::cast<mlir::DenseFPElementsAttr>(axis_stats.value()) : layer_stats;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
//===----------------------------------------------------------------------===// OpFoldResult PowOp::fold(FoldAdaptor adaptor) { auto operands = adaptor.getOperands(); auto constant_y = operands[1].dyn_cast_or_null<DenseFPElementsAttr>(); if (constant_y && constant_y.isSplat()) { APFloat y_value = constant_y.getSplatValue<APFloat>(); auto output_type = getType().cast<ShapedType>(); if (y_value.isZero() && output_type.hasStaticShape()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)