Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils_test.cc184 std::vector<AllocatorAttributes> ret_alloc_attrs; in TEST_F() local
188 &arg_alloc_attrs, &ret_alloc_attrs); in TEST_F()
194 CheckAlloc({false}, ret_alloc_attrs); in TEST_F()
Dpartitioning_utils.cc76 std::vector<AllocatorAttributes>* ret_alloc_attrs) { in UpdateArgAndRetvalMetadata() argument
118 ret_alloc_attrs->push_back(alloc_attr); in UpdateArgAndRetvalMetadata()
Dpartitioning_utils.h63 std::vector<AllocatorAttributes>* ret_alloc_attrs);
Dfunction.cc988 AllocatorAttributes ret_alloc_attrs; in RunRemote() local
990 ret_alloc_attrs.set_on_host(true); in RunRemote()
992 rets_alloc_attrs.push_back(ret_alloc_attrs); in RunRemote()
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc370 AllocatorAttributes ret_alloc_attrs; in ComputeAsync() local
372 ret_alloc_attrs.set_on_host(true); in ComputeAsync()
374 opts.rets_alloc_attrs.push_back(ret_alloc_attrs); in ComputeAsync()