Searched defs:reduce_type (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/distribution/ |
| D | distribution_collective.cc | 28 … nvinfer1::DataType data_type, ReduceMode reduce_type, in ReduceScatterWrapper()
34 … nvinfer1::DataType data_type, ReduceMode reduce_type, cudaStream_t stream, in AllReduceWrapper()
|
| D | distribution_collective_impl.cc | 37 … nvinfer1::DataType data_type, ReduceMode reduce_type, in ReduceScatterWrapper()
56 … nvinfer1::DataType data_type, ReduceMode reduce_type, cudaStream_t stream, in AllReduceWrapper()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/distribution/ |
| D | collective_wrapper.cc | 38 ncclRedOp_t reduce_type, cudaStream_t stream, const std::string &group) { in AllReduce()
48 ncclRedOp_t reduce_type, cudaStream_t stream, const std::string &group) { in ReduceScatter()
|
| D | nccl_wrapper.cc | 58 … ncclRedOp_t reduce_type, cudaStream_t stream, const std::string &group_name) { in AllReduce()
74 … ncclDataType_t data_type, ncclRedOp_t reduce_type, cudaStream_t stream, in ReduceScatter()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | bce_with_logits_loss_kernel.cc | 183 auto reduce_type = kReduceTypeMap.find(reduction)->second; in LaunchKernel() local
|