Searched refs:CHECK_KERNEL_WORKSPACE_SIZE (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | softmax_cross_entropy_with_logits_cpu_kernel.cc | 86 …CHECK_KERNEL_WORKSPACE_SIZE(workspace.size(), kSoftmaxCrossEntropyWithLogitsWorkspaceSize, kernel_… in Launch()
|
| D | sparse_softmax_cross_entropy_with_logits_cpu_kernel.cc | 117 …CHECK_KERNEL_WORKSPACE_SIZE(workspace.size(), kSparseSoftmaxCrossEntropyWithLogitsWorkspaceSize, k… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 168 …CHECK_KERNEL_WORKSPACE_SIZE(workspace.size(), kSparseApplyProximalAdagradWorkspaceSize, kernel_nam… in Launch()
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 175 CHECK_KERNEL_WORKSPACE_SIZE(workspace.size(), kSparseApplyLazyAdamWorkspaceSize, kernel_name_); in Launch()
|
| D | sparse_apply_ftrl_cpu_kernel.cc | 181 CHECK_KERNEL_WORKSPACE_SIZE(workspace.size(), kSparseApplyFtrlWorkspaceSize, kernel_name_); in Launch()
|
| D | sparse_apply_adam_cpu_kernel.cc | 205 CHECK_KERNEL_WORKSPACE_SIZE(workspace.size(), kSparseApplyAdamWorkspaceSize, kernel_name_); in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.h | 164 #define CHECK_KERNEL_WORKSPACE_SIZE(actual_size, expect_size, kernel_name) … macro
|