| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/ |
| H A D | cudnn_matmul.hpp | 105 const bool with_scratchpad = matmul_impl_->with_scratchpad(); in init() local
107 if (with_scratchpad && has_runtime_args && with_bias) { in init()
109 } else if (with_scratchpad && has_runtime_args) { in init()
115 } else if (with_bias && with_scratchpad) { in init()
117 } else if (with_scratchpad) { in init()
121 } else if (!with_scratchpad && !has_runtime_args && !with_bias) { in init()
|
| H A D | cudnn_convolution.hpp | 96 bool with_scratchpad() const { return impl_->with_scratchpad(); } in with_scratchpad() function
177 execute_convolution(ctx, pd()->with_bias(), pd()->with_scratchpad()); in execute()
182 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const;
243 bool with_scratchpad() const { return impl_->with_scratchpad(); } in with_scratchpad() function
251 ctx, pd()->with_bias(), pd()->with_scratchpad()); in execute()
254 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const;
310 bool with_scratchpad() const { return impl_->with_scratchpad(); } in with_scratchpad() function
317 ctx, pd()->with_bias(), pd()->with_scratchpad()); in execute()
320 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const;
|
| H A D | cudnn_convolution.cpp | 29 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const { in execute_convolution()
46 if (with_scratchpad) { in execute_convolution()
91 args.push_back(with_scratchpad ? sc.memory<void *>(ih, *scratch_acc) in execute_convolution()
108 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const { in execute_convolution()
123 if (with_scratchpad) { in execute_convolution()
156 args.push_back(with_scratchpad ? sc.memory<void *>(ih, *scratch_acc) in execute_convolution()
194 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const { in execute_convolution()
209 if (with_scratchpad) { in execute_convolution()
243 args.push_back(with_scratchpad ? sc.memory<void *>(ih, *scratch_acc) in execute_convolution()
|
| H A D | cudnn_matmul.cpp | 65 const auto scratchpad_size = matmul_impl_->with_scratchpad() in execute()
|
| H A D | cudnn_matmul_impl.hpp | 154 if (with_scratchpad()) { book_scratchpad(pd, dst_d.nelems()); } in init()
174 bool with_scratchpad() { return with_scratchpad_; } in with_scratchpad() function
|
| H A D | cudnn_convolution_impl.hpp | 90 bool with_scratchpad() const { return scratchpad_size > 0; } in with_scratchpad() function
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/ |
| H A D | cudnn_matmul.hpp | 117 const bool with_scratchpad = matmul_impl_->with_scratchpad(); in init() local
119 if (with_scratchpad && has_runtime_args && with_bias) { in init()
121 } else if (with_scratchpad && has_runtime_args) { in init()
127 } else if (with_bias && with_scratchpad) { in init()
129 } else if (with_scratchpad) { in init()
133 } else if (!with_scratchpad && !has_runtime_args && !with_bias) { in init()
|
| H A D | cudnn_convolution.hpp | 96 bool with_scratchpad() const { return impl_->with_scratchpad(); } in with_scratchpad() function
182 execute_convolution(ctx, pd()->with_bias(), pd()->with_scratchpad()); in execute()
187 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const;
248 bool with_scratchpad() const { return impl_->with_scratchpad(); } in with_scratchpad() function
256 ctx, pd()->with_bias(), pd()->with_scratchpad()); in execute()
259 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const;
315 bool with_scratchpad() const { return impl_->with_scratchpad(); } in with_scratchpad() function
322 ctx, pd()->with_bias(), pd()->with_scratchpad()); in execute()
325 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const;
|
| H A D | cudnn_convolution.cpp | 29 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const { in execute_convolution()
49 if (with_scratchpad) { in execute_convolution()
91 args.push_back(with_scratchpad ? sc.memory<void *>(ih, *scratch_acc) in execute_convolution()
107 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const { in execute_convolution()
122 if (with_scratchpad) { in execute_convolution()
154 args.push_back(with_scratchpad ? sc.memory<void *>(ih, *scratch_acc) in execute_convolution()
192 const exec_ctx_t &ctx, bool with_bias, bool with_scratchpad) const { in execute_convolution()
207 if (with_scratchpad) { in execute_convolution()
240 args.push_back(with_scratchpad ? sc.memory<void *>(ih, *scratch_acc) in execute_convolution()
|
| H A D | cudnn_matmul.cpp | 65 const auto scratchpad_size = matmul_impl_->with_scratchpad() in execute()
|
| H A D | cudnn_matmul_impl.hpp | 154 if (with_scratchpad()) { book_scratchpad(pd, dst_d.nelems()); } in init()
174 bool with_scratchpad() { return with_scratchpad_; } in with_scratchpad() function
|
| H A D | cudnn_convolution_impl.hpp | 93 bool with_scratchpad() const { return scratchpad_size > 0; } in with_scratchpad() function
|