Searched refs:execution_state (Results 1 – 4 of 4) sorted by relevance
72 std::unique_ptr<GraphExecutionState> execution_state; in CreateGraphExecutionState() local74 std::move(graph_def), options, &execution_state)); in CreateGraphExecutionState()75 return execution_state; in CreateGraphExecutionState()
305 execution_state = {"counter": 0, "lock": threading.Lock()}309 with execution_state["lock"]:310 execution.op_type = "OpType%d" % execution_state["counter"]311 execution_state["counter"] += 1
1609 GraphExecutionState* execution_state = nullptr; in CreateGraphs() local1622 execution_state = temp_exec_state_holder.get(); in CreateGraphs()1624 execution_state = execution_state_.get(); in CreateGraphs()1626 execution_state->BuildGraph(subgraph_options, &client_graph)); in CreateGraphs()1647 auto current_stateful_placements = execution_state->GetStatefulPlacements(); in CreateGraphs()1665 stateful_placements_ = execution_state->GetStatefulPlacements(); in CreateGraphs()1670 CopyGraph(*execution_state->full_graph(), run_state_args->graph.get()); in CreateGraphs()
232 GraphExecutionState* execution_state);1039 GraphExecutionState* execution_state) { in CheckFetches() argument1046 const Node* n = execution_state->get_node_by_name(string(id.first)); in CheckFetches()1062 const Node* n = execution_state->get_node_by_name(string(id.first)); in CheckFetches()1071 const Graph* graph = execution_state->full_graph(); in CheckFetches()