Home
last modified time | relevance | path

Searched refs:a3_size (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc2385 SmallVector<int64_t> a3_begin(input_rank * 2), a3_size(input_rank * 2); in convertStridedSliceOp() local
2391 a3_size[i * 2 + 0] = 1; in convertStridedSliceOp()
2393 a3_size[i * 2 + 0] = in convertStridedSliceOp()
2396 a3_size[i * 2 + 1] = 1; in convertStridedSliceOp()
2401 RankedTensorType::get(a3_size, input_type.getElementType()), in convertStridedSliceOp()
2403 rewriter.getI64ArrayAttr(a3_size)); in convertStridedSliceOp()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md932 vector <size_t> a1_start, a1_size, a2_shape, a3_start, a3_size, a4_shape
962 a3_size[i*2 + 0] = 1
964 a3_size[i*2 + 0] = a1_size[i] / strides[i]
966 a3_size[i*2 + 1] = 1
982 %a3_slice = tosa.SLICE(%a2_reshape) {start=a3_start, size=a3_size}