Lines Matching defs:iterators

349 /// Given an `indexingMap` and its corresponding `iterators`, returns
350 /// the positions of the iterators of type `iter` that are indexed by
358                                 ArrayRef<utils::IteratorType> iterators,
360   assert(iterators.size() == indexingMap.getNumDims());
364       if (iterators[d.getPosition()] == iter &&
387   SmallVector<utils::IteratorType> iterators(map.getNumDims(), red);
390       iterators[dim.getPosition()] = par;
391   return iterators;
407                          ArrayRef<utils::IteratorType> iterators) {
409       findPermutationsIndexingOperand(indexingMaps[0], iterators, par);
411       findPermutationsIndexingOperand(indexingMaps[1], iterators, par);
413       findPermutationsIndexingOperand(indexingMaps[2], iterators, par);
415   // A & C - B are the iterators involved in an outer-product along A (the LHS).
419   // B & C - A are the iterators involved in an outer-product along B (the RHS).
430       findPermutationsIndexingOperand(indexingMaps[0], iterators, red);
432       findPermutationsIndexingOperand(indexingMaps[1], iterators, red);
460   auto iterators = inferIteratorsFromOutMap(indexingMaps[2]);
461   if (failed(iterators))
463   return inferContractionDimsImpl(indexingMaps, iterators.value());
718   // unConvolvedDims & outputDims - filterDims are the batch iterators.
723   // convolvedDims & outputDims are the output image iterators.
727   // filterDims & outputDims - unConvolvedDims are the output channel iterators.
732   // filterDims & outputDims & unConvolvedDims are the depth iterators.
741   // convolvedDims & filterReducedDims are the filter loop iterators.
745   // unConvolvedDims & filterReducedDims are the input channel iterators.
996     return "expected all iterators used to access outputs to be parallel";
998     return "expected all iterators not used to access outputs to be reduction";