| /third_party/boost/libs/fiber/src/ |
| D | scheduler.cpp | 112 algo_{ new algo::round_robin() } { in scheduler()
143 algo_->notify(); in dispatch()
158 context * ctx = algo_->pick_next(); in dispatch()
185 algo_->suspend_until( suspend_time); in dispatch()
209 algo_->awakened( ctx); in schedule()
232 algo_->notify(); in schedule_from_remote()
259 return algo_->pick_next()->suspend_with_cc(); in terminate()
275 algo_->pick_next()->resume( ctx); in yield()
294 algo_->pick_next()->resume(); in wait_until()
319 algo_->pick_next()->resume( lk); in wait_until()
[all …]
|
| D | properties.cpp | 23 BOOST_ASSERT( nullptr != algo_); in notify()
31 dynamic_cast< algo::algorithm_with_properties_base * >( algo_)-> in notify()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | softmax_grad_gpu_kernel.h | 35 algo_(CUDNN_SOFTMAX_ACCURATE), in SoftmaxGradGpuKernel()
74 … cudnnSoftmaxBackward(cudnn_handle_, algo_, mode_, &alpha, y_desc_, y_addr, y_desc_, in Launch()
96 … cudnnSoftmaxBackward(cudnn_handle_, algo_, mode_, &alpha, y_desc_, transpose_y_addr, in Launch()
132 algo_ = CUDNN_SOFTMAX_LOG; in Init()
136 algo_ = CUDNN_SOFTMAX_ACCURATE; in Init()
209 cudnnSoftmaxAlgorithm_t algo_; variable
|
| D | conv2d_grad_filter_gpu_kernel.h | 42 algo_(CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0), in ConvGradFilterGpuBkwKernel()
94 algo_, work_space, workspace_size_, &beta, dw_desc_, dw), in Launch()
100 …utionBackwardFilter(cudnn_handle_, &alpha, x_desc_, x, dy_desc_, dy, conv_desc_, algo_, work_space, in Launch()
257 … dw_desc_, algo_, reinterpret_cast<size_t *>(&workspace_size_)), in InitSizeLists()
264 …nvolutionBackwardFilterWorkspaceSize(cudnn_handle_, x_desc_, dy_desc_, conv_desc_, dw_desc_, algo_, in InitSizeLists()
295 algo_ = perf_results.algo; in SelectAlgorithm()
301 … CUDNN_CONVOLUTION_BWD_FILTER_SPECIFY_WORKSPACE_LIMIT, 0, &algo_), in SelectAlgorithm()
306 algo_ = CUDNN_CONVOLUTION_BWD_FILTER_ALGO_1; in SelectAlgorithm()
363 cudnnConvolutionBwdFilterAlgo_t algo_; variable
|
| D | conv2d_grad_input_gpu_kernel.h | 44 algo_(CUDNN_CONVOLUTION_BWD_DATA_ALGO_0), in ConvGradInputGpuBkwKernel()
86 …olutionBackwardData(cudnn_handle_, &alpha, w_desc_, w, dy_desc_, dy, conv_desc_, algo_, work_space, in Launch()
99 …olutionBackwardData(cudnn_handle_, &alpha, w_desc_, w, dy_desc_, dy, conv_desc_, algo_, work_space, in Launch()
267 algo_, &workspace_size_), in InitSizeLists()
274 … cudnn_handle_, w_desc_, dy_desc_, conv_desc_, dx_desc_, algo_, &workspace_size_), in InitSizeLists()
310 algo_ = perf_results.algo; in SelectAlgorithm()
316 … CUDNN_CONVOLUTION_BWD_DATA_SPECIFY_WORKSPACE_LIMIT, 0, &algo_), in SelectAlgorithm()
321 algo_ = CUDNN_CONVOLUTION_BWD_DATA_ALGO_1; in SelectAlgorithm()
387 cudnnConvolutionBwdDataAlgo_t algo_; variable
|
| D | softmax_gpu_kernel.h | 36 algo_(CUDNN_SOFTMAX_ACCURATE), in SoftmaxGpuKernel()
67 … cudnnSoftmaxForward(cudnn_handle_, algo_, mode_, &alpha, input_descriptor_, in Launch()
93 …cudnnSoftmaxForward(cudnn_handle_, algo_, mode_, &alpha, input_descriptor_, transpose_input_addr, … in Launch()
126 algo_ = CUDNN_SOFTMAX_LOG; in Init()
130 algo_ = CUDNN_SOFTMAX_ACCURATE; in Init()
298 cudnnSoftmaxAlgorithm_t algo_; variable
|
| D | conv3d_grad_input_gpu_kernel.h | 57 …olutionBackwardData(cudnn_handle_, &alpha, w_desc_, w, dy_desc_, dy, conv_desc_, algo_, work_space, in Launch()
66 …olutionBackwardData(cudnn_handle_, &alpha, w_desc_, w, dy_desc_, dy, conv_desc_, algo_, work_space, in Launch()
190 algo_ = CUDNN_CONVOLUTION_BWD_DATA_ALGO_0; in ResetResource()
264 algo_, &workspace_size_), in InitSizeLists()
271 … cudnn_handle_, w_desc_, dy_desc_, conv_desc_, dx_desc_, algo_, &workspace_size_), in InitSizeLists()
309 algo_ = perf_results.algo; in SelectAlgorithm()
372 cudnnConvolutionBwdDataAlgo_t algo_; variable
|
| D | conv3d_grad_filter_gpu_kernel.h | 71 algo_, work_space, workspace_size_, &beta, dw_desc_, dw), in Launch()
77 …utionBackwardFilter(cudnn_handle_, &alpha, x_desc_, x, dy_desc_, dy, conv_desc_, algo_, work_space, in Launch()
286 … dw_desc_, algo_, reinterpret_cast<size_t *>(&workspace_size_)), in InitSizeLists()
293 …nvolutionBackwardFilterWorkspaceSize(cudnn_handle_, x_desc_, dy_desc_, conv_desc_, dw_desc_, algo_, in InitSizeLists()
332 algo_ = perf_results.algo; in SelectAlgorithm()
334 algo_ = CUDNN_CONVOLUTION_BWD_FILTER_ALGO_1; in SelectAlgorithm()
397 cudnnConvolutionBwdFilterAlgo_t algo_; variable
|
| D | conv3d_transpose_gpu_kernel.h | 62 … input_padded, conv_desc_, algo_, work_space, workspace_size_, &beta_, in Launch()
79 conv_desc_, algo_, work_space, workspace_size_, &beta_, in Launch()
88 … input_desc_, input_addr, conv_desc_, algo_, work_space, in Launch()
223 algo_ = CUDNN_CONVOLUTION_BWD_DATA_ALGO_0; in ResetResource()
329 padded_descriptor_, algo_, &workspace_size_), in InitSizeLists()
342 … stride_padded_descriptor_, algo_, &workspace_size_), in InitSizeLists()
349 output_desc_, algo_, &workspace_size_), in InitSizeLists()
388 algo_ = perf_results.algo; in SelectAlgorithm()
525 cudnnConvolutionBwdDataAlgo_t algo_; variable
|
| D | softmax_cross_entropy_with_logits_gpu_kernel.h | 36 algo_(CUDNN_SOFTMAX_ACCURATE), in SoftmaxCrossEntropyWithLogitsGpuKernel()
69 … cudnnSoftmaxForward(cudnn_handle_, algo_, mode_, &alpha, logits_descriptor_, logits_addr, &beta, in Launch()
185 cudnnSoftmaxAlgorithm_t algo_; variable
|
| D | sparse_softmax_cross_entropy_with_logits_gpu_kernel.h | 36 algo_(CUDNN_SOFTMAX_ACCURATE), in SparseSoftmaxCrossEntropyWithLogitsGpuKernel()
68 … cudnnSoftmaxForward(cudnn_handle_, algo_, mode_, &alpha, logits_descriptor_, logits_addr, &beta, in Launch()
184 cudnnSoftmaxAlgorithm_t algo_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | matmul_gpu_kernel.h | 80 compute_type, algo_), in Launch()
87 … &beta, output_addr, dtype_c_, ldc, stride_c, batch_, compute_type, algo_), in Launch()
109 algo_ = CUBLAS_GEMM_DEFAULT_TENSOR_OP; in Init()
165 algo_ = CUBLAS_GEMM_DEFAULT; in ResetResource()
199 cublasGemmAlgo_t algo_; variable
|
| D | update_thor_gradient.h | 96 CUDA_R_32F, algo_), in Launch()
128 gradient_size.batch_h * gradient_size.batch_w, CUDA_R_32F, algo_), in Launch()
252 cublasGemmAlgo_t algo_ = CUBLAS_GEMM_DEFAULT; variable
|
| /third_party/boost/boost/fiber/ |
| D | properties.hpp | 40 algo::algorithm * algo_{ nullptr }; member in boost::fibers::fiber_properties
65 algo_ = algo; in set_algorithm()
|
| D | scheduler.hpp | 89 algo::algorithm::ptr_t algo_; member in boost::fibers::scheduler
|