◆ CPUArgReductionEngine() [1/2]
◆ CPUArgReductionEngine() [2/2]
open3d::core::kernel::CPUArgReductionEngine::CPUArgReductionEngine |
( |
const Indexer & |
indexer | ) |
|
|
inline |
◆ LaunchArgReductionKernelTwoPass()
template<typename scalar_t , typename func_t >
static void open3d::core::kernel::CPUArgReductionEngine::LaunchArgReductionKernelTwoPass |
( |
const Indexer & |
indexer, |
|
|
func_t |
reduce_func, |
|
|
scalar_t |
identity |
|
) |
| |
|
inlinestatic |
Create num_threads workers to compute partial arg reductions and then reduce to the final results. This only applies to arg reduction op with one output.
◆ LaunchArgReductionParallelDim()
template<typename scalar_t , typename func_t >
static void open3d::core::kernel::CPUArgReductionEngine::LaunchArgReductionParallelDim |
( |
const Indexer & |
indexer, |
|
|
func_t |
reduce_func, |
|
|
scalar_t |
identity |
|
) |
| |
|
inlinestatic |
◆ operator=()
◆ Run()
template<typename func_t , typename scalar_t >
void open3d::core::kernel::CPUArgReductionEngine::Run |
( |
const func_t & |
reduce_func, |
|
|
scalar_t |
identity |
|
) |
| |
|
inline |
The documentation for this class was generated from the following file: