Searched refs:epsilon_attr (Results 1 – 3 of 3) sorted by relevance
66 Value MaterializeEpsilon(Operation* op, FloatAttr epsilon_attr, in MaterializeEpsilon() argument71 if (epsilon_attr.getType() != fp_type) { in MaterializeEpsilon()74 APFloat epsilon_float = epsilon_attr.getValue(); in MaterializeEpsilon()86 epsilon_attr = b.getFloatAttr(fp_type, epsilon_float); in MaterializeEpsilon()91 DenseElementsAttr::get(scalar_type, {epsilon_attr.cast<Attribute>()}); in MaterializeEpsilon()
54 const string epsilon_attr = is_fused ? "epsilon" : "variance_epsilon"; in GetScaleAndOffsetValues() local74 const float variance_epsilon = batch_norm_node.attr().at(epsilon_attr).f(); in GetScaleAndOffsetValues()
1305 auto epsilon_attr = in matchAndRewrite() local1308 rewriter.create<tosa::ConstOp>(op->getLoc(), epsilon_type, epsilon_attr); in matchAndRewrite()1365 auto epsilon_attr = in matchAndRewrite() local1368 rewriter.create<tosa::ConstOp>(op->getLoc(), epsilon_type, epsilon_attr); in matchAndRewrite()