Lines Matching defs:inputs
75 SmallVector<Value> inputs;
79 inputs.push_back(fromVals[idx++]); // The sparse tensor to disassemble
89 inputs.push_back(fromVals[idx++]);
93 mem = builder.create<sparse_tensor::ToPositionsOp>(loc, inputs[0],
96 mem = builder.create<sparse_tensor::ToCoordinatesOp>(loc, inputs[0],
99 mem = builder.create<sparse_tensor::ToValuesOp>(loc, inputs[0]);
104 inputs.push_back(extraVals[extra++]);
113 // Assemble multiple inputs into a single sparse tensor.
114 auto a = builder.create<sparse_tensor::AssembleOp>(loc, rtp, inputs);
122 builder.create<sparse_tensor::DisassembleOp>(loc, retTypes, inputs);
187 // Only sparse inputs or outputs need a wrapper method.
213 // Convert inputs.
214 SmallVector<Value> inputs;
216 ValueRange(), inputs, /*extra=*/0, /*isIn=*/true, directOut);
222 inputs);