Searched refs:size_split (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 139 Value input_value, SmallVectorImpl<int32_t>& size_split, int32_t axis);
|
| D | legalize_common.cc | 2098 Value input_value, SmallVectorImpl<int32_t>& size_split, int32_t axis) { in convertSplitVOp() argument
2128 for (int i = 0; i < size_split.size(); i++) { in convertSplitVOp()
2129 size_split_sum += size_split[i]; in convertSplitVOp()
2136 for (int i = 0; i < size_split.size(); i++) { in convertSplitVOp()
2144 size_vals.push_back(size_split[i]); in convertSplitVOp()
2162 curr_split_start += size_split[i]; in convertSplitVOp()
|
| D | legalize_tf.cc | 1572 SmallVector<int32_t> size_split; in matchAndRewrite() local
1580 size_split.push_back(size_split_elems.getValues<IntegerAttr>()[i].getInt()); in matchAndRewrite()
1593 tf_splitv_op.value(), size_split, axis); in matchAndRewrite()
|
| D | legalize_tfl.cc | 2191 SmallVector<int32_t> size_split; in matchAndRewrite() local
2199 size_split.push_back(size_split_elems.getValues<APInt>()[i].getSExtValue()); in matchAndRewrite()
2214 tfl_splitv_op.value(), size_split, axis); in matchAndRewrite()
|
| /external/pytorch/torch/distributed/tensor/_ops/ |
| D | _tensor_ops.py | 774 def size_split(N, i): function
781 size_split(input_spec.shape[dim], split_size_or_sections)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2445 int64_t size_split = size_split_attr.getValue().getSExtValue(); in verify() local
2446 size_splits.push_back(size_split); in verify()
2447 if (size_split >= 0) { in verify()
2448 total_size_splits += size_split; in verify()
2451 if (size_split < -1) in verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 863 Value lower_splitv_op(Value %value, vector <size_t> size_split, size_t axis)
869 for (int32 i = 0; i < size_split.size(); i++) {
875 size_vals.push_back(size_split[i])
884 curr_split_start += size_split[i]
|