Searched refs:axis_val (Results 1 – 5 of 5) sorted by relevance
2337 int64_t axis_val = axes_elems.getValue<IntegerAttr>(i).getInt(); in convertReduceOpCommon() local2338 if (axis_val < 0) axis_val += input_rank; in convertReduceOpCommon()2339 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in convertReduceOpCommon()2341 shape_vec[axis_val] = 1; in convertReduceOpCommon()2532 int64_t axis_val = axes_elems.getValue<IntegerAttr>(i).getInt(); in convertReduceMeanOp() local2533 if (axis_val < 0) axis_val += input_rank; in convertReduceMeanOp()2534 num_elems_on_reduced_axis *= input_type.getShape()[axis_val]; in convertReduceMeanOp()
1959 int64_t axis_val = axis_elems.getValue<IntegerAttr>(i).getInt(); in matchAndRewrite() local1960 if (axis_val < 0) axis_val += input_rank; in matchAndRewrite()1961 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite()
2773 int64_t axis_val = axis_elems.getValue<IntegerAttr>(i).getInt(); in matchAndRewrite() local2774 if (axis_val < 0) axis_val += input_rank; in matchAndRewrite()2775 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite()
148 int64_t axis_val = axis.getInt(); in InferOneHotOpType() local151 if (axis_val < -1) return unranked_ty; in InferOneHotOpType()157 if (axis_val == -1) axis_val = shape.size(); in InferOneHotOpType()164 shape.insert(shape.begin() + axis_val, depth_val); in InferOneHotOpType()
651 size_t axis_val = positive_axis(%axis.as_constant()[i])652 … %output = tosa.REVERSE(%output) {axis=%axis_val} : (tensor<%tensor.type>) -> tensor<%tensor.type>