Home
last modified time | relevance | path

Searched refs:ret_alloc_attrs_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc674 &comp_data->arg_alloc_attrs_, &comp_data->ret_alloc_attrs_)); in InstantiateMultiDevice()
712 DCHECK(comp_data.ret_alloc_attrs_.size() == comp_data.ret_indices_.size()); in GetOutputDevices()
727 comp_data.ret_alloc_attrs_[j].on_host() ? nullptr : target_device; in GetOutputDevices()
780 opts_copy.rets_alloc_attrs = comp_data.ret_alloc_attrs_; in RunMultiDevice()
Dprocess_function_library_runtime.h173 std::vector<AllocatorAttributes> ret_alloc_attrs_; member