Searched defs:get_pools (Results 1 – 3 of 3) sorted by relevance
185 def get_pools(dlname, direction=None): function
106 std::vector<std::vector<int64_t>> get_pools(const Tensor& indices, const IntArrayRef& sizes, const … in get_pools() function
450 def get_pools(self, block): member in AllocationPools