Searched defs:AutogradMeta (Results 1 – 3 of 3) sorted by relevance
205 struct TORCH_API AutogradMeta : public c10::AutogradMetaInterface { struct213 // associated with this AutogradMeta (and the Tensor it corresponds to) argument221 mutable std::shared_ptr<ForwardGrad> fw_grad_;233 std::vector<std::unique_ptr<FunctionPreHook>> hooks_;234 std::shared_ptr<hooks_list> cpp_hooks_list_;240 std::unique_ptr<PostAccumulateGradHook> post_acc_grad_hooks_ = nullptr;254 uint32_t output_nr_;261 mutable std::mutex mutex_;266 void set_requires_grad(bool requires_grad, at::TensorImpl* self_impl) final { in set_requires_grad()296 AutogradMeta( argument[all …]
91 TORCH_LIBRARY_IMPL(_, AutogradMeta, m) { in TORCH_LIBRARY_IMPL() argument
221 AutogradMeta = auto() variable in DispatchKey