- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for getBatchGroupCountAttrName (0.36 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
SmallVector<NamedAttribute> converted_attrs; for (auto attr : op->getAttrs()) { if (attr.getName() == op.getFeatureGroupCountAttrName() || attr.getName() == op.getBatchGroupCountAttrName()) { converted_attrs.push_back(attr); } else if (attr.getName() == op.getDimensionNumbersAttrName()) { attr.setValue(xla::ConvertConvDimensionNumbers(dnums, &rewriter));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0)