Searched refs:GetNodeStateOrCreateIt (Results 1 – 2 of 2) sorted by relevance
471 auto& curr_node_state = GetNodeStateOrCreateIt(curr_node); in Init()510 auto& input_node_state = GetNodeStateOrCreateIt(input_node); in Init()536 auto& input_node_state = GetNodeStateOrCreateIt(input_node); in Init()727 auto& send_node_state = GetNodeStateOrCreateIt(send); in CreateSendRecv()733 auto& recv_node_state = GetNodeStateOrCreateIt(recv); in CreateSendRecv()777 NodeState& SchedulerState::GetNodeStateOrCreateIt(const NodeDef* node) { in GetNodeStateOrCreateIt() function in tensorflow::grappler::SchedulerState
404 NodeState& GetNodeStateOrCreateIt(const NodeDef* node);