Searched refs:is_cached_tensor (Results 1 – 5 of 5) sorted by relevance
157 !at::caching::is_cached_tensor(new_grad._indices()) && in accumulateGrad()158 !at::caching::is_cached_tensor(new_grad._values()) && in accumulateGrad()159 !at::caching::is_cached_tensor(new_grad)); in accumulateGrad()
19 bool is_cached_tensor(const at::Tensor& t) { in is_cached_tensor() function44 return t.use_count() - (is_cached_tensor(t) ? 1 : 0); in adjusted_use_count()
14 TORCH_API bool is_cached_tensor(const at::Tensor& t);
127 !at::caching::is_cached_tensor(arg)); in cached_cast()
1841 return at::caching::is_cached_tensor(t); in initModule()