Home
last modified time | relevance | path

Searched refs:core_retval_nums (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3017 std::vector<std::vector<int>> core_retval_nums(num_cores_per_replica); in BuildExecuteNodes() local
3023 core_retval_nums[core].push_back(i); in BuildExecuteNodes()
3026 core_retval_nums[core].push_back(i); in BuildExecuteNodes()
3030 core_retval_nums[core].push_back(i); in BuildExecuteNodes()
3060 for (int output : core_retval_nums[core]) { in BuildExecuteNodes()
3279 int num_outputs = core_retval_nums[core].size(); in BuildExecuteNodes()
3282 replica * num_retvals_per_replica + core_retval_nums[core][i]; in BuildExecuteNodes()
3283 const auto& sharding = retval_shardings[core_retval_nums[core][i]]; in BuildExecuteNodes()
3285 int retval_index = core_retval_nums[core][i]; in BuildExecuteNodes()
3339 core_retval_nums[core_id][core_retval_index])}); in BuildExecuteNodes()
[all …]