Searched refs:VariableInput (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 319 struct VariableInput { struct
329 std::vector<VariableInput>* variables); argument
375 static Status BuildVariableReads(absl::Span<const VariableInput> variables,
400 absl::Span<const VariableInput> variables, Node* control_successor,
538 Node* control_after, absl::Span<const VariableInput> variable_nodes,
|
| D | distributed_tpu_rewrite_pass.cc | 2473 std::vector<VariableInput>* variables) { in FindVariableInputs()
2504 variables->push_back(VariableInput{input_edges[i]->src(), in FindVariableInputs()
2515 variables->push_back(VariableInput{ in FindVariableInputs()
2582 absl::Span<const VariableInput> variables, Node* control_predecessor, in BuildVariableReads()
2635 absl::Span<const VariableInput> variables, Node* control_successor, in BuildVariableWrites()
4099 Node* control_after, absl::Span<const VariableInput> variable_nodes, in DealWithConstantsAndVariables()
4301 std::vector<VariableInput> variable_inputs; in RewriteTPUReplicateNode()
|