Searched refs:ret_alloc_attrs_ (Results 1 – 2 of 2) sorted by relevance
674 &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()
173 std::vector<AllocatorAttributes> ret_alloc_attrs_; member