Searched refs:a1_size (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 2310 SmallVector<int64_t> a1_begin(input_rank), a1_size(input_rank); in convertStridedSliceOp() local
2329 a1_size[i] = -1; in convertStridedSliceOp()
2336 a1_size[i] = end[i] - a1_begin[i]; in convertStridedSliceOp()
2337 if (end[i] < 0) a1_size[i] += input_shape[i]; in convertStridedSliceOp()
2342 a1_size[i] = 1; in convertStridedSliceOp()
2350 RankedTensorType::get(a1_size, input_type.getElementType()), input_value, in convertStridedSliceOp()
2351 rewriter.getI64ArrayAttr(a1_begin), rewriter.getI64ArrayAttr(a1_size)); in convertStridedSliceOp()
2375 a2_shape[i * 2 + 0] = a1_size[i] == -1 ? -1 : a1_size[i] / abs(strides[i]); in convertStridedSliceOp()
2394 (a1_size[i] == -1) ? -1 : (a1_size[i] / abs(strides[i])); in convertStridedSliceOp()
2411 ((a1_size[i] == -1) ? -1 : (a1_size[i] / abs(strides[i])))); in convertStridedSliceOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 932 vector <size_t> a1_start, a1_size, a2_shape, a3_start, a3_size, a4_shape
953 a1_size[i] = end[i] - begin[i]
955 a2_shape[i*2 + 0] = a1_size[i] / strides[i]
964 a3_size[i*2 + 0] = a1_size[i] / strides[i]
971 a4_shape.push_back((a1_size[i] / strides[i]))
976 %a1_slice = tosa.SLICE(%input) {start=a1_start, size=a1_size}
|