Searched refs:input_shape_attr (Results 1 – 1 of 1) sorted by relevance
3341 DenseIntElementsAttr input_shape_attr; in matchAndRewrite() local3342 if (!matchPattern(op.shape(), m_Constant(&input_shape_attr))) in matchAndRewrite()3370 int64_t input_dim = (*(input_shape_attr.begin() + i)).getSExtValue(); in matchAndRewrite()4121 DenseIntElementsAttr input_shape_attr; in matchAndRewrite() local4122 if (!matchPattern(op.input_sizes(), m_Constant(&input_shape_attr)) || in matchAndRewrite()4123 input_shape_attr.getType().getRank() != 1) in matchAndRewrite()4126 auto input_shape = input_shape_attr.getValues<int32_t>(); in matchAndRewrite()