Searched refs:a4_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1188 SmallVector<int64_t, 2> a4_shape(input_rank); in convertBatchToSpaceNDOp() local
1190 a4_shape[0] = input_shape[0] / block_num_elems; in convertBatchToSpaceNDOp()
1192 a4_shape[1 + i] = input_shape[i + 1] * block_shape[i]; in convertBatchToSpaceNDOp()
1195 a4_shape[1 + block_rank + i] = input_shape[block_rank + 1 + i]; in convertBatchToSpaceNDOp()
1200 RankedTensorType::get(ArrayRef<int64_t>(a4_shape), in convertBatchToSpaceNDOp()
1202 a2_transpose_a1_op.getResult(), rewriter.getI64ArrayAttr(a4_shape)); in convertBatchToSpaceNDOp()
1223 a4_size_vals[i] = a4_shape[i] - crops[i - 1].first - crops[i - 1].second; in convertBatchToSpaceNDOp()
2070 SmallVector<int64_t, 2> a4_shape; in convertStridedSliceOp() local
2104 if (new_axis_mask & (1 << i)) a4_shape.push_back(1); in convertStridedSliceOp()
2105 a4_shape.push_back((a1_size[i] / strides[i])); in convertStridedSliceOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 367 vector <size_t> a4_shape = {%input.shape[0],
374 …utput = tosa.RESHAPE(%a3_transpose) {new_shape=a4_shape} : (tensor<%a3_transpose.type>) -> tensor<…
776 vector <size_t> a4_shape(%input.rank)
777 a4_shape[0] = batch_size * block_num_elements
780 a4_shape[i + 1] = %a1_pad.shape[i + 1] / %block.as_constant()[i]
784 a4_shape[1 + block_rank + i] = %input.shape[1 + block_rank + i]
787 …%output = tosa.RESHAPE(%a3_reshape) {new_shape=a4_shape} : (tensor<%a3_reshape.type>) -> tensor<%o…
809 vector <size_t> a4_shape = {%input.shape[0],
813 …tput = tosa.RESHAPE(%a3_transpose) {new_shape=%a4_shape} : (tensor<%a3_transpose.type>) -> tensor<…
921 vector <size_t> a1_start, a1_size, a2_shape, a3_start, a3_size, a4_shape
[all …]
|