Lines Matching defs:tensors

29 // SparseTensorLoopEmiter class, manages sparse tensors and helps to
30 // generate loop structure to (co)-iterate sparse tensors.
57   /// Optional callback function to setup dense output tensors when
68   // subscript expressions on sparse tensors.
82   /// Takes an array of input tensors, which the generated loops will
89   initialize(ValueRange tensors, StringAttr loopTag = nullptr,
95       ValueRange tensors, StringAttr loopTag = nullptr, bool hasOutput = false,
101   /// for iterating over the tensors.
138   // Still need a way to specify the lvl for non-annotated tensors though,
141   /// Emits a co-iteration loop over a set of tensors.
183   /// Gets the total number of manifest tensors (excluding the synthetic
185   unsigned getNumManifestTensors() const { return tensors.size(); }
187   /// Gets the total number of tensors that loopEmitter is operating on.
189     // Manifest tensors with one synthetic tensor at the end.
194   TensorId getSynTensorId() const { return tensors.size(); }
253   // the set of tensors levels that the loop is iterating over.
285   /// Generates instructions to compute the coordinate of tensors[tid][lvl]
388   /// loops that operates on sparse tensors more easily.
400   /// Input and (optional) output tensors.
401   std::vector<Value> tensors;