Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dtriplet_margin_loss_impl.cu216 … const std::string reduction, const bool swap, const bool need_broadcast, in CalTripletMarginLoss()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Deinsum_helper.h172 bool need_broadcast = false; in Mul() local