Home
last modified time | relevance | path

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

/external/llvm-project/mlir/docs/Dialects/
DTOSA.md78 vector <size_t> begin_vals, size_vals
82 begin_vals.push_back(slice_size * i)
85 begin_vals.push_back(0)
89 …%output[i] = tosa.SLICE(%value) {start=begin_vals, size=size_vals} (tensor<%value.type>) -> tensor…
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc321 SmallVector<int64_t, 4> begin_vals, size_vals, shape_vals; in convertUnpackOp() local
325 begin_vals.push_back(i); in convertUnpackOp()
328 begin_vals.push_back(0); in convertUnpackOp()
334 ArrayAttr begin = rewriter.getI64ArrayAttr(begin_vals); in convertUnpackOp()
1880 SmallVector<int64_t, 4> begin_vals, size_vals; in convertSplitOp() local
1884 begin_vals.push_back(slice_size * i); in convertSplitOp()
1887 begin_vals.push_back(0); in convertSplitOp()
1892 ArrayAttr begin = rewriter.getI64ArrayAttr(begin_vals); in convertSplitOp()
1960 SmallVector<int64_t, 4> begin_vals, size_vals; in convertSplitVOp() local
1964 begin_vals.push_back(curr_split_start); in convertSplitVOp()
[all …]
Dlegalize_tf.cc1421 SmallVector<int64_t, 4> begin_vals, size_vals; in matchAndRewrite() local
1429 begin_vals.push_back(begin_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
1441 ArrayAttr begin = rewriter.getI64ArrayAttr(begin_vals); in matchAndRewrite()
Dlegalize_tfl.cc1941 SmallVector<int64_t, 4> begin_vals, size_vals; in matchAndRewrite() local
1949 begin_vals.push_back(begin_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
1954 ArrayAttr begin = rewriter.getI64ArrayAttr(begin_vals); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md828 vector <size_t> begin_vals, size_vals
832 begin_vals.push_back(slice_size * i)
835 begin_vals.push_back(0)
839 …%output[i] = tosa.SLICE(%value) {start=begin_vals, size=size_vals} (tensor<%value.type>) -> tensor…
859 vector <size_t> begin_vals, size_vals
863 begin_vals.push_back(curr_split_start)
866 begin_vals.push_back(0)
871 …%output[i] = tosa.SLICE(%value) {start=begin_vals, size=size_vals} (tensor<%value.type>) -> tensor…
1009 vector <size_t> begin_vals, size_vals, shape_vals
1011 begin_vals.push_back(i)
[all …]