Searched refs:node_parameters (Results 1 – 2 of 2) sorted by relevance
1833 absl::flat_hash_map<Node*, Parameter*>& node_parameters) { in CollectBufferParametersToUpsize() argument1844 node_parameters[this] = parameter.get(); in CollectBufferParametersToUpsize()1848 for (auto& [node, parameter] : node_parameters) { in CollectBufferParametersToUpsize()2242 absl::flat_hash_map<Node*, Parameter*> node_parameters; in CollectBufferParametersToUpsize() local2244 snapshot->CollectBufferParametersToUpsize(node_parameters); in CollectBufferParametersToUpsize()2247 node->CollectBufferParametersToUpsize(node_parameters); in CollectBufferParametersToUpsize()2249 return node_parameters; in CollectBufferParametersToUpsize()2602 absl::flat_hash_map<Node*, Parameter*> node_parameters = in UpsizeBuffers() local2611 for (auto& [node, parameter] : node_parameters) { in UpsizeBuffers()2629 for (auto& [node, parameter] : node_parameters) { in UpsizeBuffers()
489 absl::flat_hash_map<Node*, Parameter*>& node_parameters);