Searched refs:batch_dimensions_ (Results 1 – 1 of 1) sorted by relevance
399 batch_dimensions_.axes.push_back(dim); in DotDimensionsInfo()400 batch_dimensions_.sizes.push_back(type.getDimSize(dim)); in DotDimensionsInfo()410 llvm::count(batch_dimensions_.axes, dim) > 0) { in DotDimensionsInfo()418 const DimensionVector &batch_dimensions() const { return batch_dimensions_; } in batch_dimensions()442 DimensionVector batch_dimensions_; member in mlir::TF::__anon0527096c0111::DotDimensionsInfo