Searched defs:tmp_loss (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | smooth_l1_loss_impl.cu | 50 T *loss, double *tmp_loss) { in SmoothL1LossSum()
65 half *loss, double *tmp_loss) { in SmoothL1LossSum()
80 T *loss, double *tmp_loss) { in SmoothL1LossMean()
96 half *loss, double *tmp_loss) { in SmoothL1LossMean()
111 __global__ void CastLoss(const double *tmp_loss, T *loss) { in CastLoss()
116 __global__ void CastLoss(const double *tmp_loss, half *loss) { in CastLoss()
123 … const T *prediction, const T *target, T *loss, double *tmp_loss, const uint32_t device_id, in SmoothL1Loss()
|
| D | loss_with_reduction_impl.cu | 29 __global__ void Copy(T *loss, T *tmp_loss, ReductionMode reduction, int input_size) { in Copy()
37 __global__ void AddTile(T *tmp_loss, int index) { in AddTile()
41 __global__ void PartialSum(T *tmp_loss, int stride) { in PartialSum()
319 T *loss, T *tmp_loss) { in KLDivLossKernel()
338 T *loss, T *tmp_loss, cudaStream_t stream) { in KLDivLoss()
388 … const T *input_y, const T *weight, T *loss, T *tmp_loss) { in BinaryCrossEntropyLossKernel()
423 … const T *input_y, const T *weight, T *loss, T *tmp_loss, cudaStream_t stream) { in BinaryCrossEntropyLoss()
|
| D | triplet_margin_loss_impl.cu | 103 __global__ void AddTile(float *tmp_loss, size_t index) { tmp_loss[0] += tmp_loss[index]; } in AddTile()
105 __global__ void PartialSum(float *tmp_loss, size_t stride) { in PartialSum()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/ |
| D | binary_cross_entropy.c | 21 … const float *input_y, const float *weight, float *loss, float *tmp_loss, in BinaryCrossEntropyLossKernel()
55 const float *weight, float *loss, float *tmp_loss, bool weight_defined) { in BinaryCrossEntropy()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | multi_margin_loss_cpu_kernel.cc | 112 std::vector<T> tmp_loss(batch_size); in LaunchKernelFP32AndFP64() local
175 std::vector<float> tmp_loss(batch_size); in LaunchKernelFP16() local
|
| D | binary_cross_entropy_cpu_kernel.cc | 66 std::vector<T> tmp_loss(input_size_); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | kl_div_loss_gpu_kernel.cc | 35 T *tmp_loss = GetDeviceAddress<T>(workspace, 0); in LaunchKernel() local
|
| D | binary_cross_entropy_gpu_kernel.cc | 56 T *tmp_loss = GetDeviceAddress<T>(workspace, kIndex0); in LaunchKernel() local
|