|
Open3D (C++ API)
0.20.0
|
SYCL tensor reduction kernels (sum/min/max and arg reductions). More...
#include <limits>#include <type_traits>#include "open3d/core/Dispatch.h"#include "open3d/core/Indexer.h"#include "open3d/core/SYCLContext.h"#include "open3d/core/Tensor.h"#include "open3d/core/kernel/Reduction.h"#include "open3d/utility/Logging.h"Namespaces | |
| namespace | open3d |
| namespace | open3d::core |
| namespace | open3d::core::kernel |
Functions | |
| void | open3d::core::kernel::ReductionSYCL (const Tensor &src, Tensor &dst, const SizeVector &dims, bool keepdim, ReductionOpCode op_code) |
SYCL tensor reduction kernels (sum/min/max and arg reductions).
Uses Indexer for strided/broadcast layouts. SYCLReductionEngine picks a work-group-per-output or blocked sycl::reduction path; SYCLArgReductionEngine uses a local tree reduction for argmin/argmax.