Searched refs:args_alloc_attrs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 386 opts.args_alloc_attrs.push_back(arg_alloc_attrs); in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 1042 std::vector<AllocatorAttributes> args_alloc_attrs, rets_alloc_attrs; in RunRemote() local
1043 args_alloc_attrs.reserve(fbody->arg_types.size()); in RunRemote()
1051 args_alloc_attrs.push_back(arg_alloc_attrs); in RunRemote()
1070 device_context, args_alloc_attrs, rendezvous, remote_args, in RunRemote()
|
| D | process_function_library_runtime.cc | 1006 opts_copy.args_alloc_attrs = comp_data.arg_alloc_attrs_; in RunMultiDevice()
1319 args.local_args, device_context, opts.args_alloc_attrs, in RunInternal()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 705 std::vector<AllocatorAttributes> args_alloc_attrs; member
|
| D | function.cc | 1726 " args_alloc_attrs=", AllocatorAttributesToString(args_alloc_attrs), in DebugString()
|