Searched refs:out_dimensions_ (Results 1 – 1 of 1) sorted by relevance
413 out_dimensions_.axes.push_back(dim); in DotDimensionsInfo()414 out_dimensions_.sizes.push_back(type.getDimSize(dim)); in DotDimensionsInfo()424 const DimensionVector &out_dimensions() const { return out_dimensions_; } in out_dimensions()436 return std::accumulate(out_dimensions_.sizes.begin(), in FlattenedOutDimensionSize()437 out_dimensions_.sizes.end(), 1, in FlattenedOutDimensionSize()446 DimensionVector out_dimensions_; member in mlir::TF::__anon0527096c0111::DotDimensionsInfo