Searched refs:retval_types (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 281 DataTypeVector* retval_types);
308 const DataTypeVector& retval_types,
434 const DataTypeVector& retval_types,
532 DataTypeVector* retval_types, ParameterInfo* params_info);
|
| D | distributed_tpu_rewrite_pass.cc | 1664 DataTypeVector* retval_types) { in GetComputationForTPUReplicateOp() argument
1674 *retval_types = fbody->ret_types; in GetComputationForTPUReplicateOp()
1915 const DataTypeVector& retval_types, in AssignArgsAndRetvalsToCores() argument
1924 std::vector<Node*> retvals(retval_types.size()); in AssignArgsAndRetvalsToCores()
2116 retval_types, retval_shapes); in AssignArgsAndRetvalsToCores()
2223 num_cores_per_replica, params_info, arg_types, arg_shapes, retval_types, in AssignArgsAndRetvalsToCores()
2891 const DataTypeVector& retval_types, in BuildExecuteNodes() argument
2902 const int num_retvals_per_replica = retval_types.size(); in BuildExecuteNodes()
3061 core_retval_types.push_back(retval_types[output]); in BuildExecuteNodes()
3975 DataTypeVector* retval_types, ParameterInfo* params_info) { in GetIOTypes() argument
[all …]
|