Home
last modified time | relevance | path

Searched defs:ndims_ (Results 1 – 14 of 14) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/compute/
H A Ddispatch.hpp135 int ndims_ = 0; member in dnnl::impl::gpu::compute::dispatch_t
/dports/math/onednn/oneDNN-2.5.1/src/gpu/compute/
H A Ddispatch.hpp135 int ndims_ = 0; member in dnnl::impl::gpu::compute::dispatch_t
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/
H A Dcudnn_resampling_impl.hpp70 int ndims_; member
H A Dcudnn_reorder_impl.hpp46 int ndims_; member
H A Dcudnn_pooling_impl.hpp188 int ndims_, kernel_ndims_; member
H A Dcudnn_inner_product_impl.hpp117 int ndims_; member
H A Dcudnn_batch_normalization_impl.hpp163 int ndims_, nchannels_; member
/dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/
H A Dcudnn_resampling_impl.hpp70 int ndims_; member
H A Dcudnn_reorder_impl.hpp46 int ndims_; member
H A Dcudnn_pooling_impl.hpp188 int ndims_, kernel_ndims_; member
H A Dcudnn_inner_product_impl.hpp115 int ndims_; member
H A Dcudnn_batch_normalization_impl.hpp163 int ndims_, nchannels_; member
/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dgemm_inner_product_utils.hpp84 int ndims_; member
/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dtensor.hpp986 int ndims_; member in dnnl::impl::gpu::jit::layout_t