Searched refs:block_size_attr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 119 IntegerAttr block_size_attr,
126 IntegerAttr block_size_attr,
|
| D | legalize_common.cc | 1671 IntegerAttr block_size_attr, in convertSpaceToDepthOp() argument
1703 if (!block_size_attr) { // This is a required parameter in convertSpaceToDepthOp()
1709 block_size.assign(2, block_size_attr.getInt()); in convertSpaceToDepthOp()
1760 IntegerAttr block_size_attr, in convertDepthToSpaceOp() argument
1789 if (!block_size_attr) { // This is a required parameter in convertDepthToSpaceOp()
1795 block_size.assign(2, block_size_attr.getInt()); in convertDepthToSpaceOp()
|
| D | legalize_tfl.cc | 2370 auto block_size_attr = tfl_s2d_op.block_sizeAttr(); in matchAndRewrite() local
2372 rewriter, op, tfl_s2d_op.getResult(), tfl_s2d_op.input(), block_size_attr, in matchAndRewrite()
2386 auto block_size_attr = tfl_d2s_op.block_sizeAttr(); in matchAndRewrite() local
2388 rewriter, op, tfl_d2s_op.getResult(), tfl_d2s_op.input(), block_size_attr, in matchAndRewrite()
|