Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/
Dhcom_all_to_all_v.cc72 auto recv_tensor = outputs[0]; in Launch() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc1076 auto recv_tensor = TensorConstructUtils::CreateZerosTensor(kFloat16, {1}); in InsertReceive() local
1653 auto recv_tensor = TensorConstructUtils::CreateZerosTensor(kFloat16, {1}); in FetchRecv() local