File tree 1 file changed +2
-2
lines changed
torch/csrc/jit/serialization
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -414,7 +414,7 @@ class GraphEncoder {
414
414
static constexpr size_t ParamSizeThresholdForExternalStorage = 1024 ;
415
415
};
416
416
417
- onnx::TensorProto_DataType ATenTypeToOnnxType (at::ScalarType at_type) {
417
+ static onnx::TensorProto_DataType ATenTypeToOnnxType (at::ScalarType at_type) {
418
418
switch (at_type) {
419
419
case at::kDouble :
420
420
return onnx::TensorProto_DataType_DOUBLE;
@@ -459,7 +459,7 @@ onnx::TensorProto_DataType ATenTypeToOnnxType(at::ScalarType at_type) {
459
459
}
460
460
}
461
461
462
- onnx::AttributeProto_AttributeType ATenAttributeKindToOnnxAttributeType (
462
+ static onnx::AttributeProto_AttributeType ATenAttributeKindToOnnxAttributeType (
463
463
AttributeKind at_kind,
464
464
const jit::Symbol name) {
465
465
switch (at_kind) {
You can’t perform that action at this time.
0 commit comments