Searched refs:tmpAttr (Results 1 – 2 of 2) sorted by relevance
504 auto tmpAttr = tf_avgpool_op.data_formatAttr(); in matchAndRewrite() local505 if (tmpAttr && tmpAttr.getValue().str() != "NHWC") return failure(); in matchAndRewrite()511 auto tmpAttr = tf_avgpool_op.strides(); in matchAndRewrite() local512 if (!tmpAttr) { in matchAndRewrite()516 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite()517 int64_t stride_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite()522 auto tmpAttr = tf_avgpool_op.ksize(); in matchAndRewrite() local523 if (!tmpAttr) { in matchAndRewrite()527 int64_t kernel_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite()528 int64_t kernel_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite()[all …]
1557 auto tmpAttr = tfl_concat_op.axisAttr(); in matchAndRewrite() local1558 if (!tmpAttr) { in matchAndRewrite()1559 tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()1561 axis_attr = tmpAttr; in matchAndRewrite()2009 auto tmpAttr = tfl_pack_op.axisAttr(); in matchAndRewrite() local2010 if (!tmpAttr) tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()2011 axis_attr = tmpAttr; in matchAndRewrite()2031 auto tmpAttr = tfl_unpack_op.axisAttr(); in matchAndRewrite() local2032 if (!tmpAttr) tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()2033 axis_attr = tmpAttr; in matchAndRewrite()