Home
last modified time | relevance | path

Searched defs:negative_broadcast (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dtriplet_margin_loss_cpu_kernel.cc326 std::vector<T> negative_broadcast(numelements_); in realtype_broadcast_task() local
400 std::vector<T> negative_broadcast(numelements_); in realtype_broadcast_compute() local
610 std::vector<T> negative_broadcast(numelements_); in complextype_broadcast_task() local
690 std::vector<T> negative_broadcast(numelements_); in complextype_broadcast_compute() local
810 std::vector<T> &negative_broadcast, in realtype_swap()
828std::vector<T> &negative_broadcast, std::vector<T> &calculate_swap, in complextype_swap()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dtriplet_margin_loss_impl.cu212 … T *positive_broadcast, T *negative_broadcast, S *output, float *tem_output, in CalTripletMarginLoss()