Searched refs:shape_vals (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1474 SmallVector<int64_t, 2> shape_vals({num_batch, num_elems}); in matchAndRewrite() local
1477 ArrayRef<int64_t>(shape_vals), input_type.getElementType()); in matchAndRewrite()
1480 rewriter.getI64ArrayAttr(shape_vals)); in matchAndRewrite()
1583 SmallVector<int64_t, 8> shape_vals; in matchAndRewrite() local
1585 shape_vals.push_back(output_type.getShape()[i]); in matchAndRewrite()
1587 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(shape_vals); in matchAndRewrite()
|
| D | legalize_common.cc | 321 SmallVector<int64_t, 4> begin_vals, size_vals, shape_vals; in convertUnpackOp() local
330 shape_vals.push_back(transposed_input_shape[j]); in convertUnpackOp()
345 RankedTensorType::get(ArrayRef<int64_t>(shape_vals), in convertUnpackOp()
347 a2_slice_op.getResult(), rewriter.getI64ArrayAttr(shape_vals)); in convertUnpackOp()
350 ArrayRef<int64_t>(shape_vals), transposed_input_type.getElementType())); in convertUnpackOp()
|
| D | legalize_tf.cc | 665 SmallVector<int64_t, 8> shape_vals; in matchAndRewrite() local
667 shape_vals.push_back(output_type.getShape()[i]); in matchAndRewrite()
669 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(shape_vals); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1009 vector <size_t> begin_vals, size_vals, shape_vals
1017 shape_vals.push_back(transposed_value.shape[j])
1021 …tosa.RESHAPE(%slice) {new_shape=shape_vals} {begin=begin_vals, size=size_vals} (tensor<%slice.type…
|