Searched refs:expand_axis (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 73 Value stb_block_shape, Value bts_block_shape, int64_t expand_axis,
114 int64_t expand_axis = -1; in matchAndRewrite() local
127 expand_axis = in matchAndRewrite()
132 if (expand_axis < 0) { in matchAndRewrite()
134 expand_axis += 4; in matchAndRewrite()
149 if (squeeze_axis != expand_axis) { in matchAndRewrite()
199 stb_op.block_shape(), bts_op.block_shape(), expand_axis, rewriter); in matchAndRewrite()
270 expand_shape.insert(expand_shape.begin() + expand_axis, 1); in matchAndRewrite()
281 conv_result_shape.insert(conv_result_shape.begin() + expand_axis, 1); in matchAndRewrite()
310 Value stb_block_shape, Value bts_block_shape, int64_t expand_axis, in ExtractDilationsAttrFromBlockShape() argument
[all …]
|