- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for IsValueInitValue (0.14 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
DenseElementsAttr operand_init; if (!matchPattern(reduce_op.getInitValues().front(), m_Constant(&operand_init))) return failure(); if (!IsValueInitValue(operand_init)) return failure(); // Check that the iota init is zero. DenseElementsAttr iota_init; if (!matchPattern(reduce_op.getInitValues().back(), m_Constant(&iota_init))) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tflite_legalize_hlo.cc
: public ConvertReduceOpToArgMinMax<TFL::ReduceMaxOp, TFL::ArgMaxOp, TFL::ReduceAnyOp, true> { public: using ConvertReduceOpToArgMinMax::ConvertReduceOpToArgMinMax; bool IsValueInitValue(const DenseElementsAttr& attr) const override { auto element_type = attr.getType().getElementType(); if (attr.getNumElements() != 1 || !element_type.isIntOrFloat()) return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0)