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()
|