Searched refs:per_host_var_copies (Results 1 – 1 of 1) sorted by relevance
2797 absl::flat_hash_map<string, std::vector<NodeOut>>* per_host_var_copies, in CreateOrGetPerHostVariableCopy() argument2799 auto it = per_host_var_copies->find(host_cpu_device); in CreateOrGetPerHostVariableCopy()2800 if (it != per_host_var_copies->end()) { in CreateOrGetPerHostVariableCopy()2880 (*per_host_var_copies)[host_cpu_device] = std::move(index_mapping); in CreateOrGetPerHostVariableCopy()3040 absl::flat_hash_map<string, std::vector<NodeOut>> per_host_var_copies; in BuildExecuteNodes() local3246 num_cores_per_replica, arg_shapes, &per_host_var_copies, in BuildExecuteNodes()