ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Eigen::internal Namespace Reference

Data Structures

struct  gemm_pack_rhs< Scalar, Index, TensorContractionSubMapper< Scalar, Index, Rhs, TensorEvaluator< const TensorReshapingOp< NewDimension, const TensorImagePatchOp< Rows, Cols, ArgType > >, Device >, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment >, nr, ColMajor, false, false >
 
struct  gemm_pack_rhs< Scalar, Index, TensorContractionSubMapper< Scalar, Index, Rhs, TensorEvaluator< const TensorReshapingOp< NewDimension, const TensorImagePatchOp< Rows, Cols, ArgType > >, Device >, nocontract_t, contract_t, 2, inner_dim_contiguous, inner_dim_reordered, Alignment >, nr, ColMajor, false, false >
 
struct  gemm_pack_rhs< Scalar, Index, TensorContractionSubMapper< Scalar, Index, Rhs, TensorEvaluator< const TensorReshapingOp< NewDimension, const TensorImagePatchOp< Rows, Cols, ArgType > >, Device >, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >, nr, ColMajor, false, false >
 
class  TensorContractionInputMapper< Scalar_, Index, Side, TensorEvaluator< const TensorReshapingOp< NewDimension, const TensorImagePatchOp< Rows, Cols, ArgType > >, Device >, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >
 
class  TensorContractionSubMapper< Scalar, Index, Side, TensorEvaluator< const TensorReshapingOp< NewDimension, const TensorImagePatchOp< Rows, Cols, ArgType > >, Device >, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >
 
class  TensorEvaluatorHasPartialPacket
 

Functions

template<typename Packet >
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE std::enable_if< unpacket_traits< Packet >::masked_load_available, typenameunpacket_traits< Packet >::mask_t >::type mask (int from, int to)
 

Function Documentation

◆ mask()

template<typename Packet >
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE std::enable_if< unpacket_traits< Packet >::masked_load_available, typenameunpacket_traits< Packet >::mask_t >::type Eigen::internal::mask ( int  from,
int  to 
)

Definition at line 75 of file eigen_convolution_helpers.h.

76{
77 const Index packet_size = internal::unpacket_traits<Packet>::size;
78 eigen_assert(0 <= from && to <= (packet_size + 1) && from < to);
79
80 using Mask = typename internal::unpacket_traits<Packet>::mask_t;
81 const Mask mask_max = std::numeric_limits<Mask>::max();
82
83 return (mask_max >> (packet_size - to)) ^ (mask_max >> (packet_size - from));
84}