MCPcopy Create free account
hub / github.com/Tencent/embedx / GetXBlockInputs

Function GetXBlockInputs

src/model/encoder/instance_encoder.cc:22–30  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

20}
21
22std::vector<GraphNode*> GetXBlockInputs(const std::string& name, int depth) {
23 std::vector<GraphNode*> inst_nodes;
24 for (int i = 0; i < depth; ++i) {
25 auto* inst_node = new InstanceNode(
26 name + std::to_string(i), Shape(BATCH_PLACEHOLDER, 0), TENSOR_TYPE_CSR);
27 inst_nodes.emplace_back(inst_node);
28 }
29 return inst_nodes;
30}
31
32// Y_UNSUPVISED_NAME
33GraphNode* GetYUnsup(const std::string& name, int label_size) {

Callers 8

GraphSageEncoderFunction · 0.85
InitGraphMethod · 0.85
InitGraphMethod · 0.85
InitGraphMethod · 0.85
InitGraphMethod · 0.85
InitGraphMethod · 0.85
GraphEncoderMethod · 0.85
InitGraphMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected