Searched refs:args_alloc_attrs (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | function_ops.cc | 357 opts.args_alloc_attrs.reserve(input_dtypes_.size()); in ComputeAsync()
361 opts.args_alloc_attrs.push_back(arg_alloc_attrs); in ComputeAsync()
|
| /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
1060 args_alloc_attrs.reserve(fbody->arg_types.size()); in RunRemote()
1068 args_alloc_attrs.push_back(arg_alloc_attrs); in RunRemote()
1087 device_context, args_alloc_attrs, rendezvous, remote_args, in RunRemote()
|
| H A D | process_function_library_runtime.cc | 1295 opts_copy.args_alloc_attrs = comp_data.arg_alloc_attrs; in RunMultiDeviceSync()
1390 opts_copy.args_alloc_attrs = comp_data.arg_alloc_attrs; in RunMultiDeviceAsync()
1814 local_args, device_context, opts.args_alloc_attrs, in RunInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | function.h | 863 std::vector<AllocatorAttributes> args_alloc_attrs; member
|
| H A D | function.cc | 1830 " args_alloc_attrs=", AllocatorAttributesToString(args_alloc_attrs), in DebugString()
|