Home
last modified time | relevance | path

Searched refs:size_vals (Results 1 – 5 of 5) sorted by relevance

/external/llvm-project/mlir/docs/Dialects/
DTOSA.md78 vector <size_t> begin_vals, size_vals
83 size_vals.push_back(slice_size)
86 size_vals.push_bac(%value.shape[j])
89 …t[i] = tosa.SLICE(%value) {start=begin_vals, size=size_vals} (tensor<%value.type>) -> tensor<size_…
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc321 SmallVector<int64_t, 4> begin_vals, size_vals, shape_vals; in convertUnpackOp() local
326 size_vals.push_back(1); in convertUnpackOp()
329 size_vals.push_back(transposed_input_shape[j]); in convertUnpackOp()
335 ArrayAttr size = rewriter.getI64ArrayAttr(size_vals); in convertUnpackOp()
339 RankedTensorType::get(ArrayRef<int64_t>(size_vals), in convertUnpackOp()
1880 SmallVector<int64_t, 4> begin_vals, size_vals; in convertSplitOp() local
1885 size_vals.push_back(slice_size); in convertSplitOp()
1888 size_vals.push_back(input_shape[j]); in convertSplitOp()
1893 ArrayAttr size = rewriter.getI64ArrayAttr(size_vals); in convertSplitOp()
1896 ArrayRef<int64_t>(size_vals), result_type.getElementType())); in convertSplitOp()
[all …]
Dlegalize_tf.cc1421 SmallVector<int64_t, 4> begin_vals, size_vals; in matchAndRewrite() local
1435 size_vals.push_back(size_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
1437 size_vals.assign(output_type.getShape().begin(), in matchAndRewrite()
1442 ArrayAttr size = rewriter.getI64ArrayAttr(size_vals); in matchAndRewrite()
Dlegalize_tfl.cc1941 SmallVector<int64_t, 4> begin_vals, size_vals; in matchAndRewrite() local
1952 size_vals.push_back(size_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
1955 ArrayAttr size = rewriter.getI64ArrayAttr(size_vals); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md828 vector <size_t> begin_vals, size_vals
833 size_vals.push_back(slice_size)
836 size_vals.push_bac(%value.shape[j])
839 …t[i] = tosa.SLICE(%value) {start=begin_vals, size=size_vals} (tensor<%value.type>) -> tensor<size_…
859 vector <size_t> begin_vals, size_vals
864 size_vals.push_back(size_split[i])
867 size_vals.push_back(input.shape[j])
871 …t[i] = tosa.SLICE(%value) {start=begin_vals, size=size_vals} (tensor<%value.type>) -> tensor<size_…
1009 vector <size_t> begin_vals, size_vals, shape_vals
1012 size_vals.push_back(1)
[all …]