Searched refs:node_parallelism_ (Results 1 – 1 of 1) sorted by relevance
145 if (node_parallelism_.contains(node)) { in Get()147 return node_parallelism_.at(node); in Get()158 node_parallelism_[node] = parameter_pair->second.get(); in Get()163 absl::flat_hash_map<const Node*, Parameter*> node_parallelism_; member in tensorflow::data::model::__anon3a9504c50111::NodeParallelismParameters