Lines Matching defs:tensors

116 LoopEmitter::LoopEmitter(ValueRange tensors, StringAttr loopTag, bool hasOutput,
120   initialize(tensors, loopTag, hasOutput, isSparseOut, numLoops, dimGetter);
136   // tensors array (len == numManifestTensor).
137   this->tensors.assign(ts.begin(), ts.end());
165       const Value t = tensors[tid];
166       // a scalar or 0-dimension tensors
204   Value tensor = tensors[t];
240     const Value tensor = tryFoldTensors(tensors[t]);
251     // input tensors. Sparse inputs use sparse primitives to obtain the values.
256       // Non-annotated dense tensors.
273       // Annotated sparse tensors.
275       // tensors.
305     const Value tensor = tryFoldTensors(tensors[t]);
335   for (TensorId t = 0, e = tensors.size(); t < e; t++) {
336     auto rtp = dyn_cast<RankedTensorType>(tensors[t].getType());
358     SmallVector<SparseIterator *> lastIter(tensors.size(), nullptr);
420     // Prepares for all the tensors used in the current loop sequence.
532   // If we need to co-iterate over two sparse tensors, we need a while loop
605       Value t = tensors[tid];
629       Value t = tensors[tid];
907   // appear before normal tensors.