Searched refs:axis_val (Results 1 – 5 of 5) sorted by relevance
2676 auto axis_val = axis.getInt(); in convertReduceOpCommon() local2677 if (axis_val < 0) axis_val += input_rank; in convertReduceOpCommon()2678 axes.push_back(axis_val); in convertReduceOpCommon()2698 for (auto axis_val : axes) { in convertReduceOpCommon() local2699 if (axis_val < 0) axis_val += input_rank; in convertReduceOpCommon()2700 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in convertReduceOpCommon()2702 shape_vec[axis_val] = 1; in convertReduceOpCommon()2902 int64_t axis_val = axes_elems.getValues<IntegerAttr>()[i].getInt(); in convertReduceMeanOp() local2903 if (axis_val < 0) axis_val += input_rank; in convertReduceMeanOp()2904 num_elems_on_reduced_axis *= input_type.getShape()[axis_val]; in convertReduceMeanOp()
2058 int64_t axis_val = axis_elems.getValues<IntegerAttr>()[i].getInt(); in matchAndRewrite() local2059 if (axis_val < 0) axis_val += input_rank; in matchAndRewrite()2060 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite()
2960 int64_t axis_val = axis_elems.getValues<APInt>()[i].getSExtValue(); in matchAndRewrite() local2961 if (axis_val < 0) axis_val += input_rank; in matchAndRewrite()2962 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite()
165 int64_t axis_val = axis.getInt(); in InferOneHotOpType() local168 if (axis_val < -1) return unranked_ty; in InferOneHotOpType()174 if (axis_val == -1) axis_val = shape.size(); in InferOneHotOpType()181 shape.insert(shape.begin() + axis_val, depth_val); in InferOneHotOpType()
662 size_t axis_val = positive_axis(%axis.as_constant()[i])663 %output = tosa.REVERSE(%output) {axis=%axis_val}