mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
75c1419b46
Summary: This PR adds functions to wrap the Pickler and exposes them to the C++ API ](https://our.intern.facebook.com/intern/diff/16746451/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/23241 Pulled By: driazati Differential Revision: D16746451 fbshipit-source-id: 25ea5db4174006ce41e2e8989c8a345b82f637a7
10 lines
262 B
C
10 lines
262 B
C
#pragma once
|
|
|
|
#include <torch/csrc/api/include/torch/types.h>
|
|
#include <torch/csrc/autograd/generated/variable_factories.h>
|
|
#include <torch/csrc/jit/custom_operator.h>
|
|
#include <torch/csrc/jit/import.h>
|
|
#include <torch/csrc/jit/pickle.h>
|
|
|
|
#include <ATen/ATen.h>
|