Searched refs:rets_alloc_attrs (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | function_ops.cc | 363 opts.rets_alloc_attrs.reserve(output_dtypes_.size()); in ComputeAsync()
377 return_type_, opts.rets_alloc_attrs.size()); in ComputeAsync()
382 << opts.rets_alloc_attrs.size() in ComputeAsync()
388 opts.rets_alloc_attrs.push_back(ret_alloc_attrs); in ComputeAsync()
|
| H A D | functional_ops.cc | 410 (opts.rets_alloc_attrs.empty() || in CondResultToBool()
411 !opts.rets_alloc_attrs[0].on_host())) { in CondResultToBool()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | function.cc | 1059 std::vector<AllocatorAttributes> args_alloc_attrs, rets_alloc_attrs; in RunRemote() local
1061 rets_alloc_attrs.reserve(fbody->ret_types.size()); in RunRemote()
1075 rets_alloc_attrs.push_back(ret_alloc_attrs); in RunRemote()
1090 rets_alloc_attrs, allow_dead_tensors](const Status& status) { in RunRemote()
1106 rets_alloc_attrs, allow_dead_tensors](const Status& status) { in RunRemote()
1119 *rets, device_context, rets_alloc_attrs, rendezvous); in RunRemote()
|
| H A D | process_function_library_runtime.cc | 1296 opts_copy.rets_alloc_attrs = comp_data.ret_alloc_attrs; in RunMultiDeviceSync()
1391 opts_copy.rets_alloc_attrs = comp_data.ret_alloc_attrs; in RunMultiDeviceAsync()
1820 const std::vector<AllocatorAttributes>& rets_alloc_attrs = in RunInternal() local
1821 opts.rets_alloc_attrs; in RunInternal()
1825 device_context, rets_alloc_attrs, remote_rets, rets, in RunInternal()
1838 device_context, rets_alloc_attrs, rendezvous, in RunInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | function.h | 864 std::vector<AllocatorAttributes> rets_alloc_attrs; member
|
| H A D | function.cc | 1831 " rets_alloc_attrs=", AllocatorAttributesToString(rets_alloc_attrs), ")"); in DebugString()
|