Home
last modified time | relevance | path

Searched refs:samples_num (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/
Dlocal_node.cc70 …GetRandomSampledNeighbors(const std::vector<std::shared_ptr<Node>> &neighbors, int32_t samples_num, in GetRandomSampledNeighbors() argument
75 int32_t num = std::min(samples_num, static_cast<int32_t>(neighbors.size())); in GetRandomSampledNeighbors()
83 … const std::vector<WeightType> &weights, int32_t samples_num, in GetWeightSampledNeighbors() argument
88 for (int32_t i = 0; i < samples_num; ++i) { in GetWeightSampledNeighbors()
95 Status LocalNode::GetSampledNeighbors(NodeType neighbor_type, int32_t samples_num, SamplingStrategy… in GetSampledNeighbors() argument
98 neighbors.reserve(samples_num); in GetSampledNeighbors()
102 while (neighbors.size() < samples_num) { in GetSampledNeighbors()
103 …RETURN_IF_NOT_OK(GetRandomSampledNeighbors(itr->second.first, samples_num - neighbors.size(), &nei… in GetSampledNeighbors()
106 …RETURN_IF_NOT_OK(GetWeightSampledNeighbors(itr->second.first, itr->second.second, samples_num, &ne… in GetSampledNeighbors()
113 for (int32_t i = 0; i < samples_num; ++i) { in GetSampledNeighbors()
Dlocal_node.h60 Status GetSampledNeighbors(NodeType neighbor_type, int32_t samples_num, SamplingStrategy strategy,
86 …GetRandomSampledNeighbors(const std::vector<std::shared_ptr<Node>> &neighbors, int32_t samples_num,
90 const std::vector<WeightType> &weights, int32_t samples_num,
Dgraph_data_impl.cc226 Status GraphDataImpl::CheckSamplesNum(NodeIdType samples_num) { in CheckSamplesNum() argument
230 if ((samples_num < 1) || (samples_num > all_nodes_number)) { in CheckSamplesNum()
232 ", got " + std::to_string(samples_num); in CheckSamplesNum()
292 int32_t samples_num, std::vector<NodeIdType> *out_samples) { in NegativeSample() argument
302 if (out_samples->size() >= samples_num) { in NegativeSample()
310 …hDataImpl::GetNegSampledNeighbors(const std::vector<NodeIdType> &node_list, NodeIdType samples_num, in GetNegSampledNeighbors() argument
313 RETURN_IF_NOT_OK(CheckSamplesNum(samples_num)); in GetNegSampledNeighbors()
337 while (neg_neighbors_vec[node_idx].size() < samples_num + 1) { in GetNegSampledNeighbors()
338 …TURN_IF_NOT_OK(NegativeSample(all_nodes, shuffled_id, &start_index, exclude_nodes, samples_num + 1, in GetNegSampledNeighbors()
349 for (int32_t i = 0; i < samples_num; ++i) { in GetNegSampledNeighbors()
Dgraph_data_impl.h104 Status GetNegSampledNeighbors(const std::vector<NodeIdType> &node_list, NodeIdType samples_num,
265 … size_t *start_index, const std::unordered_set<NodeIdType> &exclude_data, int32_t samples_num,
268 Status CheckSamplesNum(NodeIdType samples_num);
Dnode.h76 …virtual Status GetSampledNeighbors(NodeType neighbor_type, int32_t samples_num, SamplingStrategy s…
Dgraph_data.h101 …ual Status GetNegSampledNeighbors(const std::vector<NodeIdType> &node_list, NodeIdType samples_num,
Dgraph_data_client.h110 Status GetNegSampledNeighbors(const std::vector<NodeIdType> &node_list, NodeIdType samples_num,
Dgraph_data_client.cc181 …ataClient::GetNegSampledNeighbors(const std::vector<NodeIdType> &node_list, NodeIdType samples_num, in GetNegSampledNeighbors() argument
190 request.add_number(static_cast<google::protobuf::int32>(samples_num)); in GetNegSampledNeighbors()
/third_party/ffmpeg/libavcodec/
Daacenc_ltp.c101 int i, samples_num = 2048; in generate_samples() local
106 samples_num = ltp->lag + 1024; in generate_samples()
108 for (i = 0; i < samples_num; i++) in generate_samples()