Searched refs:contracting_dimensions_ (Results 1 – 1 of 1) sorted by relevance
404 contracting_dimensions_.axes.push_back(dim); in DotDimensionsInfo()405 contracting_dimensions_.sizes.push_back(type.getDimSize(dim)); in DotDimensionsInfo()409 if (llvm::count(contracting_dimensions_.axes, dim) > 0 || in DotDimensionsInfo()420 return contracting_dimensions_; in contracting_dimensions()429 return std::accumulate(contracting_dimensions_.sizes.begin(), in FlattenedContractingDimensionSize()430 contracting_dimensions_.sizes.end(), 1, in FlattenedContractingDimensionSize()443 DimensionVector contracting_dimensions_; member in mlir::TF::__anon0527096c0111::DotDimensionsInfo