Function sparseToDense

tfjs-backend-wasm/src/kernels/SparseToDense.ts:46–77

(args: {
  backend: BackendWasm,
  inputs: SparseToDenseInputs,
  attrs: SparseToDenseAttrs
})

}

function sparseToDense(args: {
  backend: BackendWasm,
  inputs: SparseToDenseInputs,
  attrs: SparseToDenseAttrs
}): TensorInfo {
  const {backend, inputs, attrs} = args;
  const {sparseIndices, sparseValues, defaultValue} = inputs;
  const {outputShape} = attrs;

  const out = backend.makeOutput(outputShape, defaultValue.dtype);
  if (util.sizeFromShape(outputShape) === 0) {
    return out;
  }

  const {sliceRank, numUpdates, sliceSize, strides, outputSize} =
      backend_util.calculateShapes(sparseValues, sparseIndices, outputShape);

  const sparseIndicesId = backend.dataIdMap.get(sparseIndices.dataId).id;
  const sparseValuesId = backend.dataIdMap.get(sparseValues.dataId).id;
  const defaultValueId = backend.dataIdMap.get(defaultValue.dataId).id;

  const stridesBytes = new Uint8Array(new Int32Array(strides).buffer);

  const outId = backend.dataIdMap.get(out.dataId).id;

  wasmSparseToDense(
      sparseIndicesId, sparseValuesId, sparseValues.shape.length,
      defaultValueId, CppDType[defaultValue.dtype], sliceRank, numUpdates,
      sliceSize, stridesBytes, outputSize, outId);

  return out;
}

export const sparseToDenseConfig: KernelConfig = {
  kernelName: SparseToDense,