Searched refs:revised_shape (Results 1 – 1 of 1) sorted by relevance
540 SmallVector<int64_t, 4> revised_shape; in RewriteNewAxisMask() local545 revised_shape.emplace_back(1); in RewriteNewAxisMask()547 revised_shape.emplace_back(original_input_shape[index++]); in RewriteNewAxisMask()552 if (failed(TF::VerifyShapeOfReshapeOp(revised_shape))) return failure(); in RewriteNewAxisMask()554 const int dim_size = revised_shape.size(); in RewriteNewAxisMask()561 rewriter.getI32IntegerAttr(static_cast<int32_t>(revised_shape[i])); in RewriteNewAxisMask()567 revised_shape, original_input_type.getElementType()); in RewriteNewAxisMask()