Home
last modified time | relevance | path

Searched refs:CudnnTensorDescriptor (Results 1 – 7 of 7) sorted by relevance

/dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/cuda/
H A Dcudnn.h50 class CudnnTensorDescriptor {
52 CudnnTensorDescriptor();
53 explicit CudnnTensorDescriptor(const Array& arr);
55 ~CudnnTensorDescriptor();
57 CudnnTensorDescriptor(const CudnnTensorDescriptor&) = delete;
58 …CudnnTensorDescriptor(CudnnTensorDescriptor&& other) noexcept : desc_{other.desc_} { other.desc_ =… in CudnnTensorDescriptor() function
59 CudnnTensorDescriptor& operator=(const CudnnTensorDescriptor&) = delete;
60 CudnnTensorDescriptor& operator=(CudnnTensorDescriptor&&) = delete;
H A Dcuda_conv.h61 …void AddBias(CudnnHandle& handle, const CudnnTensorDescriptor& y_desc, const Array& y, const Array…
65 const CudnnTensorDescriptor& x_desc,
70 const CudnnTensorDescriptor& y_desc,
79 const CudnnTensorDescriptor& x_desc,
82 const CudnnTensorDescriptor& y_desc,
89 const CudnnTensorDescriptor& x_desc,
91 const CudnnTensorDescriptor& gy_desc,
H A Dcudnn.cc89 CudnnTensorDescriptor::CudnnTensorDescriptor() { CheckCudnnError(cudnnCreateTensorDescriptor(&desc_… in CudnnTensorDescriptor() function in chainerx::cuda::cuda_internal::CudnnTensorDescriptor
91 CudnnTensorDescriptor::~CudnnTensorDescriptor() { in ~CudnnTensorDescriptor()
97 CudnnTensorDescriptor::CudnnTensorDescriptor(const Array& arr) : CudnnTensorDescriptor{} { in CudnnTensorDescriptor() function in chainerx::cuda::cuda_internal::CudnnTensorDescriptor
111 Dtype CudnnTensorDescriptor::GetDtype() const { in GetDtype()
H A Dcuda_conv.cc112 CudnnTensorDescriptor b_desc{b_cont}; in AddBias()
125 const CudnnTensorDescriptor& x_desc, in FindConvolutionForwardAlgorithm()
130 const CudnnTensorDescriptor& y_desc, in FindConvolutionForwardAlgorithm()
176 const CudnnTensorDescriptor& x_desc, in FindConvolutionBackwardDataAlgorithm()
179 const CudnnTensorDescriptor& y_desc, in FindConvolutionBackwardDataAlgorithm()
223 const CudnnTensorDescriptor& x_desc, in FindConvolutionBackwardFilterAlgorithm()
321 CudnnTensorDescriptor x_desc{x_cont}; in Conv()
322 CudnnTensorDescriptor y_desc{y}; in Conv()
428 CudnnTensorDescriptor x_desc{x_cont}; in ConvTranspose()
429 CudnnTensorDescriptor y_desc{y}; in ConvTranspose()
[all …]
/dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/cuda/cuda_device/
H A Drnn.cu261 std::vector<cuda_internal::CudnnTensorDescriptor> xs_desc; in Call()
295 cuda_internal::CudnnTensorDescriptor hx_desc{hx}; in Call()
302 cuda_internal::CudnnTensorDescriptor cx_desc{_cx}; in Call()
303 cuda_internal::CudnnTensorDescriptor hy_desc{hy}; in Call()
304 cuda_internal::CudnnTensorDescriptor cy_desc{cy}; in Call()
417 cuda_internal::CudnnTensorDescriptor hx_desc{hx}; in Call()
427 cuda_internal::CudnnTensorDescriptor cx_desc{_cx}; in Call()
428 cuda_internal::CudnnTensorDescriptor dcy_desc{_dcy}; in Call()
429 cuda_internal::CudnnTensorDescriptor dhy_desc{dhy}; in Call()
431 cuda_internal::CudnnTensorDescriptor dhx_desc{dhx}; in Call()
[all …]
H A Dbatch_norm.cc79 cuda_internal::CudnnTensorDescriptor DeriveBatchNormTensorDescriptor( in DeriveBatchNormTensorDescriptor()
80 const cuda_internal::CudnnTensorDescriptor& x_desc, cudnnBatchNormMode_t mode) { in DeriveBatchNormTensorDescriptor()
81 cuda_internal::CudnnTensorDescriptor derive_desc{}; in DeriveBatchNormTensorDescriptor()
137 cuda_internal::CudnnTensorDescriptor x_desc{ExpandToAtLeast4D(x_cont)}; in Call()
140 …cuda_internal::CudnnTensorDescriptor gamma_beta_mean_var_desc = DeriveBatchNormTensorDescriptor(x_… in Call()
255 cuda_internal::CudnnTensorDescriptor x_desc{ExpandToAtLeast4D(x_cont)}; in Call()
259 …cuda_internal::CudnnTensorDescriptor gamma_beta_mean_var_desc = DeriveBatchNormTensorDescriptor(x_… in Call()
352 cuda_internal::CudnnTensorDescriptor x_desc{ExpandToAtLeast4D(x_cont)}; in Call()
356 …cuda_internal::CudnnTensorDescriptor gamma_beta_mean_var_desc = DeriveBatchNormTensorDescriptor(x_… in Call()
H A Dpool.cu125 cuda_internal::CudnnTensorDescriptor x_desc{x_cont}; in Pool()
126 cuda_internal::CudnnTensorDescriptor out_desc{actual_out}; in Pool()
179 cuda_internal::CudnnTensorDescriptor out_desc{out_cont}; in PoolGrad()
180 cuda_internal::CudnnTensorDescriptor gout_desc{gout_cont}; in PoolGrad()
181 cuda_internal::CudnnTensorDescriptor x_desc{x_cont}; in PoolGrad()
182 cuda_internal::CudnnTensorDescriptor gx_desc{actual_gx}; in PoolGrad()