Searched refs:name_to_send (Results 1 – 2 of 2) sorted by relevance
33 std::unordered_map<string, const NodeDef*> name_to_send; in ComputeTransitiveFanin() local38 name_to_send[attr.at("tensor_name").s()] = &node; in ComputeTransitiveFanin()77 const NodeDef* send = name_to_send[attr.at("tensor_name").s()]; in ComputeTransitiveFanin()
455 std::unordered_map<string, const NodeDef*> name_to_send; in Init() local459 name_to_send[attr.at("tensor_name").s()] = &node; in Init()478 auto it = name_to_send.find(send_node_name); in Init()481 if (it != name_to_send.end()) { in Init()