Jetson Inference
DNN Vision Library
cudaVector.h File Reference
#include "cudaMath.h"
#include <type_traits>

Go to the source code of this file.

Classes

struct  cudaVectorTypeInfo< T >
 
struct  cudaVectorTypeInfo< uchar >
 
struct  cudaVectorTypeInfo< uchar3 >
 
struct  cudaVectorTypeInfo< uchar4 >
 
struct  cudaVectorTypeInfo< float >
 
struct  cudaVectorTypeInfo< float3 >
 
struct  cudaVectorTypeInfo< float4 >
 
struct  cuda_assert_false< T >
 

Vector Templates

struct cudaVectorTypeInfo< uchar__attribute__
 
template<typename T >
__host__ __device__ T make_vec (typename cudaVectorTypeInfo< T >::Base x, typename cudaVectorTypeInfo< T >::Base y, typename cudaVectorTypeInfo< T >::Base z, typename cudaVectorTypeInfo< T >::Base w)
 
template<>
__host__ __device__ uchar make_vec (uint8_t x, uint8_t y, uint8_t z, uint8_t w)
 
template<>
__host__ __device__ float make_vec (float x, float y, float z, float w)
 
template<typename T >
__host__ __device__ T cast_vec (const uchar3 &a)
 
template<typename T >
__host__ __device__ T cast_vec (const uchar4 &a)
 
template<typename T >
__host__ __device__ T cast_vec (const float3 &a)
 
template<typename T >
__host__ __device__ T cast_vec (const float4 &a)
 
template<>
__host__ __device__ uchar3 cast_vec (const uchar3 &a)
 
template<>
__host__ __device__ uchar3 cast_vec (const uchar4 &a)
 
template<>
__host__ __device__ uchar3 cast_vec (const float3 &a)
 
template<>
__host__ __device__ uchar3 cast_vec (const float4 &a)
 
template<typename T >
__device__ cudaVectorTypeInfo< T >::Base alpha (T vec, typename cudaVectorTypeInfo< T >::Base default_alpha=255)
 
template<>
__host__ __device__ uint8_t alpha (uchar3 vec, uint8_t default_alpha)
 
template<>
__host__ __device__ uint8_t alpha (uchar4 vec, uint8_t default_alpha)
 
template<>
__host__ __device__ float alpha (float3 vec, float default_alpha)
 
template<>
__host__ __device__ float alpha (float4 vec, float default_alpha)
 

Function Documentation

◆ alpha() [1/5]

template<>
__host__ __device__ float alpha ( float3  vec,
float  default_alpha 
)
inline

◆ alpha() [2/5]

template<>
__host__ __device__ float alpha ( float4  vec,
float  default_alpha 
)
inline

◆ alpha() [3/5]

template<typename T >
__device__ cudaVectorTypeInfo<T>::Base alpha ( vec,
typename cudaVectorTypeInfo< T >::Base  default_alpha = 255 
)
inline

◆ alpha() [4/5]

template<>
__host__ __device__ uint8_t alpha ( uchar3  vec,
uint8_t  default_alpha 
)
inline

◆ alpha() [5/5]

template<>
__host__ __device__ uint8_t alpha ( uchar4  vec,
uint8_t  default_alpha 
)
inline

◆ cast_vec() [1/8]

template<typename T >
__host__ __device__ T cast_vec ( const float3 &  a)
inline

◆ cast_vec() [2/8]

template<>
__host__ __device__ float4 cast_vec ( const float3 &  a)
inline

◆ cast_vec() [3/8]

template<typename T >
__host__ __device__ T cast_vec ( const float4 &  a)
inline

◆ cast_vec() [4/8]

template<>
__host__ __device__ float4 cast_vec ( const float4 &  a)
inline

◆ cast_vec() [5/8]

template<typename T >
__host__ __device__ T cast_vec ( const uchar3 &  a)
inline

◆ cast_vec() [6/8]

template<>
__host__ __device__ float4 cast_vec ( const uchar3 &  a)
inline

◆ cast_vec() [7/8]

template<typename T >
__host__ __device__ T cast_vec ( const uchar4 &  a)
inline

◆ cast_vec() [8/8]

template<>
__host__ __device__ float4 cast_vec ( const uchar4 &  a)
inline

◆ make_vec() [1/3]

template<>
__host__ __device__ float4 make_vec ( float  x,
float  y,
float  z,
float  w 
)
inline

◆ make_vec() [2/3]

template<typename T >
__host__ __device__ T make_vec ( typename cudaVectorTypeInfo< T >::Base  x,
typename cudaVectorTypeInfo< T >::Base  y,
typename cudaVectorTypeInfo< T >::Base  z,
typename cudaVectorTypeInfo< T >::Base  w 
)
inline

◆ make_vec() [3/3]

template<>
__host__ __device__ uchar4 make_vec ( uint8_t  x,
uint8_t  y,
uint8_t  z,
uint8_t  w 
)
inline

Variable Documentation

◆ __attribute__

struct cudaVectorTypeInfo< uchar > __attribute__