Searched defs:set_allocator (Results 1 – 4 of 4) sorted by relevance
57 void set_allocator(Allocator& allocator) { allocator_ = &allocator; } in set_allocator() function
43 ExecutableRunOptions& ExecutableRunOptions::set_allocator( in set_allocator() function in xla::ExecutableRunOptions
193 void set_allocator(at::Allocator* allocator) { in set_allocator() function
540 void set_allocator(boost_ptree_raw_alloc_func *af, boost_ptree_raw_free_func *ff) in set_allocator() function in boost::property_tree::detail::rapidxml::memory_pool