Classes | Public Types | Static Public Member Functions | Static Private Member Functions | List of all members
Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar > Struct Template Reference

#include <TensorContraction.h>

Classes

struct  BlockSizes
 

Public Types

typedef void * BlockMemHandle
 

Static Public Member Functions

template<typename Device >
static EIGEN_DEVICE_FUNC BlockMemHandle allocate (Device &d, const Index bm, const Index bk, const Index bn, LhsScalar **lhs_block, RhsScalar **rhs_block)
 
template<typename Device >
static EIGEN_DEVICE_FUNC BlockMemHandle allocateSlices (Device &d, const Index bm, const Index bk, const Index bn, const Index num_lhs, const Index num_rhs, const Index num_slices, std::vector< LhsScalar * > *lhs_blocks, std::vector< RhsScalar * > *rhs_blocks)
 
template<typename Device >
static EIGEN_DEVICE_FUNC void deallocate (Device &d, BlockMemHandle handle)
 

Static Private Member Functions

static EIGEN_DEVICE_FUNC BlockSizes ComputeLhsRhsBlockSizes (const Index bm, const Index bk, const Index bn)
 

Detailed Description

template<typename LhsScalar, typename RhsScalar>
struct Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar >

Definition at line 79 of file TensorContraction.h.

Member Typedef Documentation

◆ BlockMemHandle

template<typename LhsScalar , typename RhsScalar >
typedef void* Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar >::BlockMemHandle

Definition at line 80 of file TensorContraction.h.

Member Function Documentation

◆ allocate()

template<typename LhsScalar , typename RhsScalar >
template<typename Device >
static EIGEN_DEVICE_FUNC BlockMemHandle Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar >::allocate ( Device &  d,
const Index  bm,
const Index  bk,
const Index  bn,
LhsScalar **  lhs_block,
RhsScalar **  rhs_block 
)
inlinestatic

Definition at line 83 of file TensorContraction.h.

◆ allocateSlices()

template<typename LhsScalar , typename RhsScalar >
template<typename Device >
static EIGEN_DEVICE_FUNC BlockMemHandle Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar >::allocateSlices ( Device &  d,
const Index  bm,
const Index  bk,
const Index  bn,
const Index  num_lhs,
const Index  num_rhs,
const Index  num_slices,
std::vector< LhsScalar * > *  lhs_blocks,
std::vector< RhsScalar * > *  rhs_blocks 
)
inlinestatic

Definition at line 99 of file TensorContraction.h.

◆ ComputeLhsRhsBlockSizes()

template<typename LhsScalar , typename RhsScalar >
static EIGEN_DEVICE_FUNC BlockSizes Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar >::ComputeLhsRhsBlockSizes ( const Index  bm,
const Index  bk,
const Index  bn 
)
inlinestaticprivate

Definition at line 140 of file TensorContraction.h.

◆ deallocate()

template<typename LhsScalar , typename RhsScalar >
template<typename Device >
static EIGEN_DEVICE_FUNC void Eigen::internal::TensorContractionBlockMemAllocator< LhsScalar, RhsScalar >::deallocate ( Device &  d,
BlockMemHandle  handle 
)
inlinestatic

Definition at line 131 of file TensorContraction.h.


The documentation for this struct was generated from the following file:


gtsam
Author(s):
autogenerated on Fri Nov 1 2024 03:49:32