Searched defs:ReduceWorkspaceSize (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
| H A D | broadcast_reduce-inl.h | 290 size_t ReduceWorkspaceSize(Stream<cpu> *s, const mxnet::TShape& small, const OpReqType req, in ReduceWorkspaceSize() function
296 size_t ReduceWorkspaceSize(Stream<cpu> *s, const mxnet::TShape& small, const OpReqType req, in ReduceWorkspaceSize() function
|
| H A D | broadcast_reduce-inl.cuh | 662 size_t ReduceWorkspaceSize(Stream<gpu> *s, const mxnet::TShape& small, const OpReqType req, in ReduceWorkspaceSize() function
670 size_t ReduceWorkspaceSize(Stream<gpu> *s, const mxnet::TShape& small, const OpReqType req, in ReduceWorkspaceSize() function
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
| H A D | broadcast_reduce-inl.h | 290 size_t ReduceWorkspaceSize(Stream<cpu> *s, const mxnet::TShape& small, const OpReqType req, in ReduceWorkspaceSize() function
296 size_t ReduceWorkspaceSize(Stream<cpu> *s, const mxnet::TShape& small, const OpReqType req, in ReduceWorkspaceSize() function
|
| H A D | broadcast_reduce-inl.cuh | 662 size_t ReduceWorkspaceSize(Stream<gpu> *s, const mxnet::TShape& small, const OpReqType req, in ReduceWorkspaceSize() function
670 size_t ReduceWorkspaceSize(Stream<gpu> *s, const mxnet::TShape& small, const OpReqType req, in ReduceWorkspaceSize() function
|