Searched refs:batch_dim_attr (Results 1 – 1 of 1) sorted by relevance
4232 auto batch_dim_attr = rewriter.getI64IntegerAttr(batch_dim); in matchAndRewrite() local4246 /*input_batch_dimension=*/batch_dim_attr, in matchAndRewrite()4257 /*output_batch_dimension=*/batch_dim_attr, in matchAndRewrite()4439 auto batch_dim_attr = rewriter.getI64IntegerAttr(batch_dim); in matchAndRewrite() local4453 /*input_feature_dimension=*/batch_dim_attr, in matchAndRewrite()4459 /*kernel_input_feature_dimension=*/batch_dim_attr, in matchAndRewrite()