Searched defs:map_num (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | sparse_fill_empty_rows_grad_impl.cu | 23 __global__ void AssignValue(const size_t map_num, const int64_t *reverse_map, const T *grad_values,… in AssignValue()
32 __global__ void AssignValue(const size_t map_num, const int64_t *reverse_map, const Complex<float> … in AssignValue()
41 __global__ void AssignValue(const size_t map_num, const int64_t *reverse_map, const Complex<double>… in AssignValue()
50 __global__ void CmpValue(const size_t map_num, const size_t values_num, const T *grad_values, bool … in CmpValue()
71 __global__ void CmpValue(const size_t map_num, const size_t values_num, const half *grad_values, bo… in CmpValue()
92 __global__ void CmpValue(const size_t map_num, const size_t values_num, const float *grad_values, b… in CmpValue()
113 __global__ void CmpValue(const size_t map_num, const size_t values_num, const Complex<double> *grad… in CmpValue()
135 __global__ void CmpValue(const size_t map_num, const size_t values_num, const Complex<float> *grad_… in CmpValue()
199 cudaError_t CalFillRowsGrad(const size_t map_num, const size_t values_num, const int64_t *reverse_m… in CalFillRowsGrad()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/dataset/ |
| D | test_callbacks.py | 81 def verify_events(events, epoch_num, step_num, step_size=1, map_num=1, repeat=1): argument
122 def generate_expected(epoch_num, step_num, step_size=1, map_num=1, repeat=1): argument
|