Searched defs:tensor_info_t (Results 1 – 1 of 1) sorted by relevance
80 struct tensor_info_t { struct81 tensor_info_t(const std::string &node_name, uint32_t slot, uint32_t iteration, uint32_t rank_id, in tensor_info_t() function89 const std::string get_node_name() const { return node_name; } in get_node_name()90 const uint32_t get_slot() const { return slot; } in get_slot()91 const uint32_t get_iteration() const { return iteration; } in get_iteration()92 const uint32_t get_rank_id() const { return rank_id; } in get_rank_id()93 const uint32_t get_root_graph_id() const { return root_graph_id; } in get_root_graph_id()94 const bool get_is_output() const { return is_output; } in get_is_output()95 std::string node_name;96 uint32_t slot;[all …]