Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils_test.cc199 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
205 graph.get(), device_type, &arg_indices, &ret_indices, &arg_alloc_attrs, in TEST_F()
211 CheckAlloc({false}, arg_alloc_attrs); in TEST_F()
238 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
244 graph.get(), device_type, &arg_indices, &ret_indices, &arg_alloc_attrs, in TEST_F()
250 CheckAlloc({false, false, false, false, false}, arg_alloc_attrs); in TEST_F()
Dpartitioning_utils.cc79 std::vector<AllocatorAttributes>* arg_alloc_attrs, in UpdateArgAndRetvalMetadata() argument
127 if (arg_alloc_attrs != nullptr) { in UpdateArgAndRetvalMetadata()
137 arg_alloc_attrs->push_back(alloc_attr); in UpdateArgAndRetvalMetadata()
Dpartitioning_utils.h66 std::vector<AllocatorAttributes>* arg_alloc_attrs,
Dprocess_function_library_runtime.h253 std::vector<AllocatorAttributes> arg_alloc_attrs; member
Dfunction.cc1029 AllocatorAttributes arg_alloc_attrs; in RunRemote() local
1031 arg_alloc_attrs.set_on_host(true); in RunRemote()
1033 args_alloc_attrs.push_back(arg_alloc_attrs); in RunRemote()
Dprocess_function_library_runtime.cc941 &comp_data->ret_indices, &comp_data->arg_alloc_attrs, in InstantiateMultiDevice()
1128 opts_copy.args_alloc_attrs = comp_data.arg_alloc_attrs; in RunMultiDevice()
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc338 AllocatorAttributes arg_alloc_attrs; in ComputeAsync() local
339 arg_alloc_attrs.set_on_host(DataTypeAlwaysOnHost(dtype)); in ComputeAsync()
340 opts.args_alloc_attrs.push_back(arg_alloc_attrs); in ComputeAsync()