Searched refs:size_vals (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 370 SmallVector<int64_t> begin_vals, size_vals, shape_vals; in convertUnpackOp() local
375 size_vals.push_back(1); in convertUnpackOp()
378 size_vals.push_back(transposed_input_shape[j]); in convertUnpackOp()
384 ArrayAttr size = rewriter.getI64ArrayAttr(size_vals); in convertUnpackOp()
388 RankedTensorType::get(size_vals, in convertUnpackOp()
2059 SmallVector<int64_t> begin_vals, size_vals; in convertSplitOp() local
2062 size_vals.push_back(slice_type.getDimSize(j)); in convertSplitOp()
2064 size_vals[axis] = size_vals[axis] / num_split; in convertSplitOp()
2068 begin_vals[axis] = i * size_vals[axis]; in convertSplitOp()
2070 ArrayAttr size = rewriter.getI64ArrayAttr(size_vals); in convertSplitOp()
[all …]
|
| D | legalize_tf.cc | 1422 SmallVector<int64_t> begin_vals, size_vals; in matchAndRewrite() local
1436 size_vals.push_back(size_elems.getValues<IntegerAttr>()[i].getInt()); in matchAndRewrite()
1438 size_vals.assign(output_type.getShape().begin(), in matchAndRewrite()
1443 ArrayAttr size = rewriter.getI64ArrayAttr(size_vals); in matchAndRewrite()
|
| D | legalize_tfl.cc | 2046 SmallVector<int64_t> begin_vals, size_vals; in matchAndRewrite() local
2057 size_vals.push_back(size_elems.getValues<APInt>()[i].getSExtValue()); in matchAndRewrite()
2060 ArrayAttr size = rewriter.getI64ArrayAttr(size_vals); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 839 vector <size_t> begin_vals, size_vals
844 size_vals.push_back(slice_size)
847 size_vals.push_bac(%value.shape[j])
850 %output[i] = tosa.SLICE(%value) {start=begin_vals, size=size_vals}
870 vector <size_t> begin_vals, size_vals
875 size_vals.push_back(size_split[i])
878 size_vals.push_back(input.shape[j])
882 %output[i] = tosa.SLICE(%value) {start=begin_vals, size=size_vals}
1020 vector <size_t> begin_vals, size_vals, shape_vals
1023 size_vals.push_back(1)
[all …]
|