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.cc2069 SmallVector<int64_t, 2> a3_begin(input_rank * 2), a3_size(input_rank * 2); in convertStridedSliceOp() local
2097 a3_size[i * 2 + 0] = 1; in convertStridedSliceOp()
2099 a3_size[i * 2 + 0] = a1_size[i] / strides[i]; in convertStridedSliceOp()
2101 a3_size[i * 2 + 1] = 1; in convertStridedSliceOp()
2130 RankedTensorType::get(ArrayRef<int64_t>(a3_size), in convertStridedSliceOp()
2133 rewriter.getI64ArrayAttr(a3_size)); in convertStridedSliceOp()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md921 vector <size_t> a1_start, a1_size, a2_shape, a3_start, a3_size, a4_shape
951 a3_size[i*2 + 0] = 1
953 a3_size[i*2 + 0] = a1_size[i] / strides[i]
955 a3_size[i*2 + 1] = 1
971 … = tosa.SLICE(%a2_reshape) {start=a3_start, size=a3_size} : (tensor<%a2_reshape.type>) -> tensor<a…