Searched refs:a1_size (Results 1 – 2 of 2) sorted by relevance
2067 SmallVector<int64_t, 2> a1_begin(input_rank), a1_size(input_rank); in convertStridedSliceOp() local2088 a1_size[i] = end[i] - begin[i]; in convertStridedSliceOp()2090 a2_shape[i * 2 + 0] = a1_size[i] / strides[i]; in convertStridedSliceOp()2099 a3_size[i * 2 + 0] = a1_size[i] / strides[i]; in convertStridedSliceOp()2105 a4_shape.push_back((a1_size[i] / strides[i])); in convertStridedSliceOp()2115 RankedTensorType::get(ArrayRef<int64_t>(a1_size), in convertStridedSliceOp()2118 rewriter.getI64ArrayAttr(a1_size)); in convertStridedSliceOp()
921 vector <size_t> a1_start, a1_size, a2_shape, a3_start, a3_size, a4_shape942 a1_size[i] = end[i] - begin[i]944 a2_shape[i*2 + 0] = a1_size[i] / strides[i]953 a3_size[i*2 + 0] = a1_size[i] / strides[i]960 a4_shape.push_back((a1_size[i] / strides[i]))965 …1_slice = tosa.SLICE(%input) {start=a1_start, size=a1_size} : (tensor<%input.type>) -> tensor<a1_s…