createFloat32Tensor method

Pointer<Float> createFloat32Tensor(
  1. Pointer<Pointer<OrtValue>> inputTensorPointer, {
  2. required Pointer<OrtMemoryInfo> memoryInfo,
  3. required List<double> values,
})

Implementation

Pointer<Float> createFloat32Tensor(
  Pointer<Pointer<OrtValue>> inputTensorPointer, {
  required Pointer<OrtMemoryInfo> memoryInfo,
  required List<double> values,
}) {
  final sizeOfFloat32 = sizeOf<Float>();
  final inputTensorNative = calloc<Float>(values.length * sizeOfFloat32);
  final float32List = Float32List.fromList(values);
  for (var i = 0; i < values.length; i++) {
    inputTensorNative[i] = float32List[i];
  }
  final inputShape = calloc<Int64>(sizeOf<Int64>());
  inputShape[0] = values.length;
  final ptrVoid = inputTensorNative.cast<Void>();
  final status = createTensorWithDataAsOrtValue(
    inputTensorPointer,
    memoryInfo: memoryInfo,
    inputData: ptrVoid,
    inputDataLengthInBytes: values.length * sizeOfFloat32,
    inputShape: inputShape,
    inputShapeLengthInBytes: 1,
    onnxTensorElementDataType:
        ONNXTensorElementDataType.ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT.value,
  );
  if (status.isError) {
    final error =
        'Code: ${getErrorCodeMessage(status)}\n'
        'Message: ${consumeErrorMessage(status)}';
    calloc.free(inputTensorNative);
    calloc.free(inputShape);
    throw Exception(error);
  }
  calloc.free(inputShape);
  return inputTensorNative;
}