Home
last modified time | relevance | path

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/
Dtriplet_margin_loss_impl.cu26 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