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

Function GetXInput

src/model/encoder/instance_encoder.cc:16–20  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

14namespace embedx {
15
16GraphNode* GetXInput(const std::string& name) {
17 // Shape (BATCH_PLACEHOLDER = 100, ) cannot reshape to (n,3) when compiling
18 // the graph. Change X shape from (BATCH_PLACEHOLDER, ) to (-1, ).
19 return new InstanceNode(name, Shape(-1, 0), TENSOR_TYPE_CSR);
20}
21
22std::vector<GraphNode*> GetXBlockInputs(const std::string& name, int depth) {
23 std::vector<GraphNode*> inst_nodes;

Callers 15

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

Calls

no outgoing calls

Tested by

no test coverage detected