Searched refs:size_split (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 139 SmallVector<int32_t, 4>& size_split,
|
| D | legalize_common.cc | 1919 SmallVector<int32_t, 4>& size_split, in convertSplitVOp() argument
1945 for (int i = 0; i < size_split.size(); i++) { in convertSplitVOp()
1946 size_split_sum += size_split[i]; in convertSplitVOp()
1957 for (int i = 0; i < size_split.size(); i++) { in convertSplitVOp()
1965 size_vals.push_back(size_split[i]); in convertSplitVOp()
1987 curr_split_start += size_split[i]; in convertSplitVOp()
|
| D | legalize_tf.cc | 1573 SmallVector<int32_t, 4> size_split; in matchAndRewrite() local
1581 size_split.push_back(size_split_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
1594 tf_splitv_op.value(), size_split, axis); in matchAndRewrite()
|
| D | legalize_tfl.cc | 2088 SmallVector<int32_t, 4> size_split; in matchAndRewrite() local
2096 size_split.push_back(size_split_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
2111 tfl_splitv_op.value(), size_split, axis); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1953 int64_t size_split = size_split_attr.getValue().getSExtValue(); in Verify() local
1954 size_splits.push_back(size_split); in Verify()
1955 if (size_split >= 0) { in Verify()
1956 total_size_splits += size_split; in Verify()
1959 if (size_split < -1) in Verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 852 Value lower_splitv_op(Value %value, vector <size_t> size_split, size_t axis)
858 for (int i = 0; i < size_split.size(); i++) {
864 size_vals.push_back(size_split[i])
873 curr_split_start += size_split[i]
|