Searched refs:shape_vals (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1502 SmallVector<int64_t, 2> shape_vals({num_batch, num_elems}); in matchAndRewrite() local
1505 RankedTensorType::get(shape_vals, input_type.getElementType()); in matchAndRewrite()
1508 rewriter.getI64ArrayAttr(shape_vals)); in matchAndRewrite()
1637 SmallVector<int64_t> shape_vals; in matchAndRewrite() local
1644 shape_vals.resize(output_type.getRank(), -1); in matchAndRewrite()
1646 for (auto dim : shape_attr.getValues<int32_t>()) shape_vals.push_back(dim); in matchAndRewrite()
1651 if (output_type.getRank() != shape_vals.size()) return failure(); in matchAndRewrite()
1653 if (shape_vals[i] == -1) shape_vals[i] = output_type.getDimSize(i); in matchAndRewrite()
1659 for (auto val : shape_vals) in matchAndRewrite()
1663 ArrayAttr new_shape_attr = rewriter.getI64ArrayAttr(shape_vals); in matchAndRewrite()
|
| D | legalize_tf.cc | 670 SmallVector<int64_t> shape_vals; in matchAndRewrite() local
672 shape_vals.push_back(output_type.getShape()[i]); in matchAndRewrite()
674 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(shape_vals); in matchAndRewrite()
|
| D | legalize_common.cc | 370 SmallVector<int64_t> begin_vals, size_vals, shape_vals; in convertUnpackOp() local
379 shape_vals.push_back(transposed_input_shape[j]); in convertUnpackOp()
394 RankedTensorType::get(shape_vals, in convertUnpackOp()
396 a2_slice_op.getResult(), rewriter.getI64ArrayAttr(shape_vals)); in convertUnpackOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1020 vector <size_t> begin_vals, size_vals, shape_vals
1028 shape_vals.push_back(transposed_value.shape[j])
1032 … %output_arr[i] = %tosa.RESHAPE(%slice) {new_shape=shape_vals} {begin=begin_vals, size=size_vals}
|