Lines Matching full:batch

402 ///   5. Optional batch dimensions that appear in all operands are captured.
423   // A & B & C are the "batch" dimensions.
441   llvm::sort(dimensions.batch.begin(), dimensions.batch.end());
718   // unConvolvedDims & outputDims - filterDims are the batch iterators.
719   llvm::SmallDenseSet<int64_t> batch = inputExprWalker.unConvolvedDims;
720   llvm::set_intersect(batch, outputDims);
721   llvm::set_subtract(batch, filterDims);
754       SmallVector<unsigned, 2>(batch.begin(), batch.end()),
762   llvm::sort(dimensions.batch.begin(), dimensions.batch.end());
798 ///   1. Optional batch dimensions that appear in the input and filter.
880   // - Batch loop : present in output, as non-convolved in input, not present in
897       // Batch dimension.