1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
|
#pragma once
/**
* @file
* Declarations shared between import_data.cpp and export_data.cpp
*/
namespace torch {
namespace jit {
namespace mobile {
namespace internal {
/**
* The name of the mobile::Module attribute which contains saved parameters, as
* a Dict of names to Tensors. Only used for Flatbuffer serialization.
*/
// NOLINTNEXTLINE(cppcoreguidelines-avoid-c-arrays,modernize-avoid-c-arrays)
constexpr char kSavedParametersAttributeName[] = "data";
} // namespace internal
} // namespace mobile
} // namespace jit
} // namespace torch
|