Searched refs:args_alloc_attrs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 336 opts.args_alloc_attrs.reserve(input_dtypes_.size()); in ComputeAsync()
340 opts.args_alloc_attrs.push_back(arg_alloc_attrs); in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 1024 std::vector<AllocatorAttributes> args_alloc_attrs, rets_alloc_attrs; in RunRemote() local
1025 args_alloc_attrs.reserve(fbody->arg_types.size()); in RunRemote()
1033 args_alloc_attrs.push_back(arg_alloc_attrs); in RunRemote()
1052 device_context, args_alloc_attrs, rendezvous, remote_args, in RunRemote()
|
| D | process_function_library_runtime.cc | 1128 opts_copy.args_alloc_attrs = comp_data.arg_alloc_attrs; in RunMultiDevice()
1524 local_args, device_context, opts.args_alloc_attrs, in RunInternal()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 804 std::vector<AllocatorAttributes> args_alloc_attrs; member
|
| D | function.cc | 1768 " args_alloc_attrs=", AllocatorAttributesToString(args_alloc_attrs), in DebugString()
|