Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h150 int64 num_retvals_per_replica) in ParameterInfo() argument
157 num_retvals_per_replica_(num_retvals_per_replica) {} in ParameterInfo()
Ddistributed_tpu_rewrite_pass.cc2902 const int num_retvals_per_replica = retval_types.size(); in BuildExecuteNodes() local
3282 replica * num_retvals_per_replica + core_retval_nums[core][i]; in BuildExecuteNodes()