Home
last modified time | relevance | path

Searched refs:ret_alloc_attrs (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils_test.cc200 std::vector<AllocatorAttributes> ret_alloc_attrs; in TEST_F() local
206 &ret_alloc_attrs); in TEST_F()
212 CheckAlloc({false}, ret_alloc_attrs); in TEST_F()
239 std::vector<AllocatorAttributes> ret_alloc_attrs; in TEST_F() local
245 &ret_alloc_attrs); in TEST_F()
251 CheckAlloc({false, false, false, false, false}, ret_alloc_attrs); in TEST_F()
Dpartitioning_utils.cc80 std::vector<AllocatorAttributes>* ret_alloc_attrs) { in UpdateArgAndRetvalMetadata() argument
144 if (ret_alloc_attrs) { in UpdateArgAndRetvalMetadata()
154 ret_alloc_attrs->push_back(alloc_attr); in UpdateArgAndRetvalMetadata()
Dpartitioning_utils.h67 std::vector<AllocatorAttributes>* ret_alloc_attrs);
Dprocess_function_library_runtime.h256 std::vector<AllocatorAttributes> ret_alloc_attrs; member
Dprocess_function_library_runtime.cc942 &comp_data->ret_alloc_attrs)); in InstantiateMultiDevice()
1027 DCHECK(comp_data.ret_alloc_attrs.size() == comp_data.ret_indices.size()); in GetOutputDevices()
1055 comp_data.ret_alloc_attrs[j].on_host() ? host : target_device; in GetOutputDevices()
1129 opts_copy.rets_alloc_attrs = comp_data.ret_alloc_attrs; in RunMultiDevice()
Dfunction.cc1036 AllocatorAttributes ret_alloc_attrs; in RunRemote() local
1038 ret_alloc_attrs.set_on_host(true); in RunRemote()
1040 rets_alloc_attrs.push_back(ret_alloc_attrs); in RunRemote()
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc344 AllocatorAttributes ret_alloc_attrs; in ComputeAsync() local
345 ret_alloc_attrs.set_on_host(DataTypeAlwaysOnHost(dtype)); in ComputeAsync()
346 opts.rets_alloc_attrs.push_back(ret_alloc_attrs); in ComputeAsync()