Home
last modified time | relevance | path

Searched refs:maxIdx (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dmaxpool_with_argmax_impl.cu54 S maxIdx = posc*h*w + hstart*w + wstart; in MaxPoolWithArgmax() local
55 T maxData = input[inputStart+maxIdx]; in MaxPoolWithArgmax()
61 maxIdx = inputIdx; in MaxPoolWithArgmax()
67 index[pos] = maxIdx; in MaxPoolWithArgmax()
/third_party/cmsis/CMSIS/DSP/Source/StatisticsFunctions/
Darm_max_f32.c159 uint32x4_t maxIdx; in arm_max_f32() local
168 maxIdx = vdupq_n_u32(ULONG_MAX); in arm_max_f32()
229 countV = vbslq_u32(idxV, countV,maxIdx); in arm_max_f32()
Darm_min_f32.c163 uint32x4_t maxIdx; in arm_min_f32() local
169 maxIdx = vdupq_n_u32(ULONG_MAX); in arm_min_f32()
229 countV = vbslq_u32(idxV, countV,maxIdx); in arm_min_f32()
/third_party/mindspore/mindspore/lite/java/java/fl_client/src/main/java/com/mindspore/flclient/model/
DTrainModel.java277 int maxIdx = start; in getPredictLabel() local
280 maxIdx = i; in getPredictLabel()
284 return maxIdx - start; in getPredictLabel()