Searched defs:num_element (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/rl/ |
| D | discounted_return_impl.cu | 22 __global__ void DiscountedReturnKernel(const int timestep, const int num_env, const int num_element… in DiscountedReturnKernel()
42 cudaError_t DiscountedReturn(const int ×tep, const int &num_env, const int &num_element, const… in DiscountedReturn()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | matrix_diag_v3_cpu_kernel.cc | 184 size_t num_element = static_cast<size_t>(outputs[0]->size() / sizeof(T)); in LaunchKernel() local
214 size_t num_element = static_cast<size_t>(outputs[0]->size() / sizeof(T)); in DoLaunch() local
|
| D | reversev2_cpu_kernel.cc | 95 int64_t num_element = 1; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/rl/ |
| D | priority_replay_buffer_cpu_kernel.cc | 48 …size_t num_element = std::accumulate(shapes[i].begin(), shapes[i].end(), 1ULL, std::multiplies<siz… in Resize() local
114 …size_t num_element = std::accumulate(shapes[i].begin(), shapes[i].end(), 1ULL, std::multiplies<siz… in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | batchmatmul.cc | 117 uint64_t num_element = output_tensor->NumElements(); in DoCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | conv2d.cc | 89 …64_t> CheckAttrIntOrTuple(const ValuePtr &attr, const size_t start_idx, const size_t num_element) { in CheckAttrIntOrTuple()
|
| D | conv3d.cc | 181 … const size_t start_idx, const size_t num_element) { in Conv3DCheckAttrIntOrTuple()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/utils/ |
| D | check_convert_utils.cc | 1388 size_t num_element) { in CheckAttrTuple()
|