Searched refs:node_mem (Results 1 – 4 of 4) sorted by relevance
82 NodeMemProto *node_mem = graph_proto->add_node_mems(); in MemoryToPB() local83 if (node_mem == nullptr) { in MemoryToPB()87 node_mem->set_node_name(node.GetNodeName()); in MemoryToPB()88 node_mem->set_node_id(node.GetNodeId()); in MemoryToPB()90 node_mem->add_input_tensor_id(id); in MemoryToPB()93 node_mem->add_output_tensor_id(id); in MemoryToPB()96 node_mem->add_workspace_tensor_id(id); in MemoryToPB()
90 local node_mem=092 node_mem=$((total_nodes - 1))93 node_mem=$((node_mem * PAGESIZE * 128))96 limit=$((limit + 4 * PAGESIZE + total_cpus * PAGESIZE + node_mem))
296 node_mem = 0300 node_mem += size302 return node_mem
496 'src/node_mem.h',497 'src/node_mem-inl.h',