Home
last modified time | relevance | path

Searched refs:recv_type (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
Dtest_neighborexchange.py54 … send_shapes=([32, 32], [32, 16]), recv_type=ms.float32)
81 send_shapes=([32, 32],), recv_type=ms.float32)
106 send_shapes=(), recv_type=ms.float32)
128 send_shapes=([32, 16],), recv_type=ms.float32)
150 send_shapes=(), recv_type=ms.float32)
174 send_shapes=([32, 32],), recv_type=ms.float32)
203 send_shapes=([32, 32],), recv_type=ms.float32)
232 … send_shapes=([32, 32], [32, 32]), recv_type=ms.float32)
260 send_shapes=([16, 16],), recv_type=ms.float32)
286 send_shapes=([32, 16],), recv_type=ms.float32)
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel_metadata.cc76 TypeId recv_type; in HcclMetadataInfo() local
78 if (!HcomUtil::GetHcomReceiveType(kernel_node, &recv_type)) { in HcclMetadataInfo()
81 auto res = std::find(kHcclSupportTypes.begin(), kHcclSupportTypes.end(), recv_type); in HcclMetadataInfo()
83 MS_LOG(EXCEPTION) << "HcclReceive cannot support data type: " << TypeIdToType(recv_type); in HcclMetadataInfo()
100 outputs_type.push_back(recv_type); in HcclMetadataInfo()
Dhcom_all_to_all.cc38 auto recv_type = AnfAlgo::GetNodeAttr<TypePtr>(anf_node, kAttrRecvType); in Init() local
39 MS_EXCEPTION_IF_NULL(recv_type); in Init()
40 data_type_ = HcomUtil::ConvertHcclType(recv_type->type_id()); in Init()
/third_party/mindspore/mindspore/core/ops/
Dneighborexchange.cc174 auto recv_type = primitive->GetAttr(kRecvType)->cast<TypePtr>(); in InferType() local
175 MS_EXCEPTION_IF_NULL(recv_type); in InferType()
176 std::vector<TypePtr> type_vec(out_num, recv_type); in InferType()
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dall_to_all_unify_mindir_test.py38 recv_type=ms.float32)
/third_party/mindspore/mindspore/ops/operations/
Dcomm_ops.py643 def __init__(self, send_rank_ids, recv_rank_ids, recv_shapes, send_shapes, recv_type, argument
650 self.recv_type = recv_type
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_comm_ops.py368 recv_type = self.recv_type
369 …e_grad = NeighborExchange(send_rank_ids, recv_rank_ids, recv_shapes, send_shapes, recv_type, group)
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dconv2d_info.cc660 Attr recv_type = {RECV_TYPE, dtype}; in CreateNeighborExchangeAttrs() local
661 OperatorAttrs attrs = {send_ranks, recv_ranks, recv_shapes, send_shapes, recv_type}; in CreateNeighborExchangeAttrs()