Searched defs:pair_tensor (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 | 26 const T *pair_tensor[3] = {anchor, positive, negative}; in FillAndBroadcast() local
58 const T *pair_tensor[3] = {anchor, positive, negative}; in PairwiseDistance() local
122 const double *pair_tensor[3] = {anchor, positive, negative}; in PairwiseDistance() local
148 const half *pair_tensor[3] = {anchor, positive, negative}; in PairwiseDistance() local
189 const Complex<S> *pair_tensor[3] = {anchor, positive, negative}; in PairwiseDistance() local
|