Searched defs:tem_output (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | triplet_margin_loss_impl.cu | 57 … float *tem_output, const size_t n, const int64_t p, const float eps) { in PairwiseDistance()
78 …_ void PairwiseDistancePzero(const size_t *bound_list, const size_t output_size, float *tem_output, in PairwiseDistancePzero()
87 __global__ void SwapTrue(float *tem_output, const size_t output_size) { in SwapTrue()
96 __global__ void MaxReduction(float *tem_output, float *output, const size_t output_size, const floa… in MaxReduction()
112 __global__ void ReductionDivde(S *output, float *tem_output, const size_t k) { in ReductionDivde()
120 … const size_t inner_size, float *tem_output, const size_t n, const int64_t p, in PairwiseDistance()
146 … const size_t inner_size, float *tem_output, const size_t n, const int64_t p, in PairwiseDistance()
169 __global__ void MaxReduction(float *tem_output, half *output, const size_t output_size, const float… in MaxReduction()
179 __global__ void ReductionDivde(half *output, float *tem_output, const size_t k) { in ReductionDivde()
187 … const size_t inner_size, float *tem_output, const size_t n, const int64_t p, in PairwiseDistance()
[all …]
|