Searched refs:total_num (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | assessment_method.py | 25 self.total_num = 0
34 self.total_num += len(labels)
37 return self.acc_num / self.total_num * 100.0
44 self.total_num = 0
57 self.total_num += len(labels)
60 return self.acc_num / self.total_num * 100.0
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | minmax_update_impl.cu | 61 … const int total_num, const int channel_num, const float ema_decay, const bool ema, in CalMinMaxPerChannel() argument
63 int per_channel_num = total_num / channel_num; in CalMinMaxPerChannel()
70 … const int total_num, const float ema_decay, const bool ema, cudaStream_t cuda_stream) { in CalMinMaxPerLayer() argument
76 …lement(policy, thrust::device_pointer_cast(input), thrust::device_pointer_cast(input) + total_num); in CalMinMaxPerLayer()
|
| D | fake_quant_perchannel_impl.cuh | 26 void CalFakeQuantPerChannel(const float *input, float *output, const int total_num, const int chann…
30 …eQuantPerChannelGrad(const float *input, const float *gradient, float *output, const int total_num,
|
| D | fake_quant_perchannel_impl.cu | 133 …eQuantPerChannelGrad(const float *input, const float *gradient, float *output, const int total_num, in CalFakeQuantPerChannelGrad() argument
136 …lGrad<<<GET_BLOCKS(channel_num), GET_THREADS, 0, cuda_stream>>>(input, gradient, output, total_num, in CalFakeQuantPerChannelGrad()
|
| D | minmax_update_impl.cuh | 23 … const int total_num, const int channel_num, const float ema_decay, const bool ema,
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 235 size_t total_num = input_num + output_num + workspace_num; in IsAtomicNode() local
238 for (; pad_index < total_num; ++pad_index) { in IsAtomicNode()
248 if (parameters_indexs.size() < total_num) { in IsAtomicNode()
250 << " less than total num: " << total_num; in IsAtomicNode()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/ |
| D | opencl_kernel.cc | 136 …auto total_num = mem_type == lite::opencl::MemType::BUF ? img_info.ElementsNum : img_info.Elements… in PrintOutput() local
137 for (int i = 0; i < print_num && i < total_num; ++i) { in PrintOutput()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | control_node_parser.cc | 638 size_t total_num = 0; in FetchOutputSizebyCallNode() local
648 total_num += call_output_num; in FetchOutputSizebyCallNode()
653 total_num += tuple_value.size(); in FetchOutputSizebyCallNode()
655 ++total_num; in FetchOutputSizebyCallNode()
659 return total_num; in FetchOutputSizebyCallNode()
|