reference_pool.hpp File Reference#
reference_pool.hpp File Reference
#include "ck_tile/core.hpp"#include "ck_tile/host/host_tensor.hpp"#include "ck_tile/ops/pooling/kernel/pool_kernel.hpp"#include <thread>#include <cmath>Go to the source code of this file.
Namespaces | |
| namespace | ck_tile |
Functions | |
| template<typename InDataType, typename ComputeDataType, typename OutDataType, typename IndexDataType, typename ReduceOp, typename TensorShape, typename WindowShape, bool OutputIndex = false> | |
| CK_TILE_HOST void | ck_tile::reference_pool2d (const HostTensor< InDataType > &input, HostTensor< OutDataType > &output, HostTensor< IndexDataType > &output_index, PoolKernelArgs< TensorShape, WindowShape > kargs, ReduceOp reduce_op) |
| template<typename InDataType, typename ComputeDataType, typename OutDataType, typename IndexDataType, typename ReduceOp, typename TensorShape, typename WindowShape, bool OutputIndex = false> | |
| CK_TILE_HOST void | ck_tile::reference_pool3d (const HostTensor< InDataType > &input, HostTensor< OutDataType > &output, HostTensor< IndexDataType > &output_index, PoolKernelArgs< TensorShape, WindowShape > kargs, ReduceOp reduce_op) |