Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfunction_ops.cc363 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 Dfunctional_ops.cc410 (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 Dfunction.cc1059 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 Dprocess_function_library_runtime.cc1296 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 Dfunction.h864 std::vector<AllocatorAttributes> rets_alloc_attrs; member
H A Dfunction.cc1831 " rets_alloc_attrs=", AllocatorAttributesToString(rets_alloc_attrs), ")"); in DebugString()