Searched refs:variable_writes (Results 1 – 2 of 2) sorted by relevance
401 absl::Span<const VariableWrite> variable_writes, Graph* graph);440 std::vector<VariableWrite>* variable_writes, Graph* graph);
2636 absl::Span<const VariableWrite> variable_writes, Graph* graph) { in BuildVariableWrites() argument2637 CHECK_EQ(variables.size(), variable_writes.size()); in BuildVariableWrites()2639 const VariableWrite& write = variable_writes[i]; in BuildVariableWrites()2897 std::vector<VariableWrite>* variable_writes, Graph* graph) { in BuildExecuteNodes() argument2904 variable_writes->resize(num_variables); in BuildExecuteNodes()3431 VariableWrite& write = variable_writes->at(core_variable_writes[i]); in BuildExecuteNodes()3446 VariableWrite& write = variable_writes->at(core_variable_writes[i]); in BuildExecuteNodes()4336 std::vector<VariableWrite> variable_writes; in RewriteTPUReplicateNode() local4341 control_after_compilation, control_after, &variable_writes, graph)); in RewriteTPUReplicateNode()4350 variable_writes, graph)); in RewriteTPUReplicateNode()