Lines Matching full:dimensions

52 /// These dimensions are such that:
59 ///   5. Optional batch dimensions that appear in all operands are captured.
90 /// These dimensions are such that:
91 ///   1. Optional batch dimensions that appear in the input and filter.
104 ///   7. All dimensions appear only once in any given indexing map.
124 ///  `linalg.broadcast`. Returns broadcast dimensions if true.
129 ///  `linalg.transpose`. Returns permuted dimensions if true.
175 /// `dimensions` with indexes of the different kinds of dimensions when
179                            ContractionDimensions *dimensions = nullptr);
190 /// `dimensions` with indexes of the different kinds of dimensions when
196                            ConvolutionDimensions *dimensions = nullptr,