Searched refs:a4_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1131 SmallVector<int64_t> a4_shape(input_rank); in convertBatchToSpaceNDOp() local
1133 a4_shape[0] = input_shape[0]; in convertBatchToSpaceNDOp()
1134 if (a4_shape[0] != ShapedType::kDynamicSize) { in convertBatchToSpaceNDOp()
1135 a4_shape[0] /= block_num_elems; in convertBatchToSpaceNDOp()
1138 a4_shape[1 + i] = multiply_dims(input_shape[i + 1], block_shape[i]); in convertBatchToSpaceNDOp()
1141 a4_shape[1 + block_rank + i] = input_shape[block_rank + 1 + i]; in convertBatchToSpaceNDOp()
1146 RankedTensorType::get(a4_shape, result_type.getElementType()), in convertBatchToSpaceNDOp()
1147 a2_transpose_a1_op.getResult(), rewriter.getI64ArrayAttr(a4_shape)); in convertBatchToSpaceNDOp()
1168 a4_size_vals[i] = a4_shape[i] - crops[i - 1].first - crops[i - 1].second; in convertBatchToSpaceNDOp()
2406 SmallVector<int64_t> a4_shape; in convertStridedSliceOp() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 335 vector <size_t> a4_shape = {%input.shape[0],
342 %output = tosa.RESHAPE(%a3_transpose) {new_shape=a4_shape}
787 vector <size_t> a4_shape(%input.rank)
788 a4_shape[0] = batch_size * block_num_elements
791 a4_shape[i + 1] = %a1_pad.shape[i + 1] / %block.as_constant()[i]
795 a4_shape[1 + block_rank + i] = %input.shape[1 + block_rank + i]
798 %output = tosa.RESHAPE(%a3_reshape) {new_shape=a4_shape}
820 vector <size_t> a4_shape = {%input.shape[0],
824 %output = tosa.RESHAPE(%a3_transpose) {new_shape=%a4_shape}
932 vector <size_t> a1_start, a1_size, a2_shape, a3_start, a3_size, a4_shape
[all …]
|