Searched refs:arg_alloc_attrs (Results 1 – 5 of 5) sorted by relevance
183 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local188 &arg_alloc_attrs, &ret_alloc_attrs); in TEST_F()193 CheckAlloc({false}, arg_alloc_attrs); in TEST_F()
75 std::vector<AllocatorAttributes>* arg_alloc_attrs, in UpdateArgAndRetvalMetadata() argument107 arg_alloc_attrs->push_back(alloc_attr); in UpdateArgAndRetvalMetadata()
62 std::vector<AllocatorAttributes>* arg_alloc_attrs,
981 AllocatorAttributes arg_alloc_attrs; in RunRemote() local983 arg_alloc_attrs.set_on_host(true); in RunRemote()985 args_alloc_attrs.push_back(arg_alloc_attrs); in RunRemote()
363 AllocatorAttributes arg_alloc_attrs; in ComputeAsync() local365 arg_alloc_attrs.set_on_host(true); in ComputeAsync()367 opts.args_alloc_attrs.push_back(arg_alloc_attrs); in ComputeAsync()