Searched refs:name_to_send (Results 1 – 2 of 2) sorted by relevance
204 std::unordered_map<string, const NodeDef*> name_to_send; in ComputeTransitiveFanin() local209 name_to_send[attr.at("tensor_name").s()] = &node; in ComputeTransitiveFanin()246 const NodeDef* send = name_to_send[attr.at("tensor_name").s()]; in ComputeTransitiveFanin()
339 std::unordered_map<string, const NodeDef*> name_to_send; in Init() local343 name_to_send[attr.at("tensor_name").s()] = &node; in Init()362 auto it = name_to_send.find(send_node_name); in Init()365 if (it != name_to_send.end()) { in Init()