Skip to content

Boolean tensors

Houssem ROUIS requested to merge boolean_tensors into dev

Context

This merge request adds support for the Boolean type to the aidge_backend_cuda module.
This includes:

  • Add boolean data type to TensorImpl.
  • Follow the ONNX convention by setting the output of the Equal operator to bool.

Modified files

  • TensorImpl.hpp and TensorImpl.cu Add boolean data type to TensorImpl.
  • EqualImpl.hpp , EqualImpl.cpp, EqualImpl_CUDA_kernels.hpp, EqualImpl_CUDA_kernels.cu and Tests_EqualImpl.cpp Force Equal's output to be of type Boolean.
Edited by Houssem ROUIS

Merge request reports

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent