FM / utils

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils

Name	Size	Mode	Actions
auto_gil.h	1034	0644	edit dl rm
byte_order.h	2484	0644	edit dl rm
crash_handler.h	1146	0644	edit dl rm
cuda_enabled.h	154	0644	edit dl rm
cuda_lazy_init.h	974	0644	edit dl rm
disable_torch_function.h	863	0644	edit dl rm
disallow_copy.h	103	0644	edit dl rm
init.h	324	0644	edit dl rm
invalid_arguments.h	302	0644	edit dl rm
memory.h	1173	0644	edit dl rm
numpy_stub.h	399	0644	edit dl rm
object_ptr.h	1329	0644	edit dl rm
out_types.h	294	0644	edit dl rm
pybind.h	7966	0644	edit dl rm
pycfunction_helpers.h	209	0644	edit dl rm
python_arg_parser.h	30753	0644	edit dl rm
python_compat.h	2904	0644	edit dl rm
python_dispatch.h	174	0644	edit dl rm
python_numbers.h	5064	0644	edit dl rm
python_scalars.h	2928	0644	edit dl rm
python_strings.h	4592	0644	edit dl rm
python_stub.h	56	0644	edit dl rm
python_tuples.h	684	0644	edit dl rm
six.h	1425	0644	edit dl rm
structseq.h	153	0644	edit dl rm
tensor_apply.h	431	0644	edit dl rm
tensor_dtypes.h	244	0644	edit dl rm
tensor_flatten.h	2780	0644	edit dl rm
tensor_layouts.h	107	0644	edit dl rm
tensor_list.h	196	0644	edit dl rm
tensor_memoryformats.h	113	0644	edit dl rm
tensor_new.h	1807	0644	edit dl rm
tensor_numpy.h	542	0644	edit dl rm
tensor_qschemes.h	186	0644	edit dl rm
tensor_types.h	491	0644	edit dl rm
throughput_benchmark-inl.h	5210	0644	edit dl rm
throughput_benchmark.h	6868	0644	edit dl rm
variadic.h	4394	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils/byte_order.h (2484B)

#pragma once

#include <TH/THHalf.h>
#include <c10/util/BFloat16.h>
#include <torch/csrc/WindowsTorchApiMacro.h>
#include <cstddef>
#include <cstdint>

namespace torch {
namespace utils {

enum THPByteOrder {
  THP_LITTLE_ENDIAN = 0,
  THP_BIG_ENDIAN = 1
};

TORCH_API THPByteOrder THP_nativeByteOrder();

TORCH_API void THP_decodeInt16Buffer(
    int16_t* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeInt32Buffer(
    int32_t* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeInt64Buffer(
    int64_t* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeHalfBuffer(
    THHalf* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeFloatBuffer(
    float* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeDoubleBuffer(
    double* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeBoolBuffer(
    bool* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeBFloat16Buffer(
    at::BFloat16* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeComplexFloatBuffer(
    c10::complex<float>* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_decodeComplexDoubleBuffer(
    c10::complex<double>* dst,
    const uint8_t* src,
    THPByteOrder order,
    size_t len);

TORCH_API void THP_encodeInt16Buffer(
    uint8_t* dst,
    const int16_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_encodeInt32Buffer(
    uint8_t* dst,
    const int32_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_encodeInt64Buffer(
    uint8_t* dst,
    const int64_t* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_encodeFloatBuffer(
    uint8_t* dst,
    const float* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_encodeDoubleBuffer(
    uint8_t* dst,
    const double* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_encodeComplexloatBuffer(
    uint8_t* dst,
    const c10::complex<float>* src,
    THPByteOrder order,
    size_t len);
TORCH_API void THP_encodeComplexDoubleBuffer(
    uint8_t* dst,
    const c10::complex<double>* src,
    THPByteOrder order,
    size_t len);

} // namespace utils
} // namespace torch

#pragma once #include #include #include #include #include namespace torch { namespace utils { enum THPByteOrder { THP_LITTLE_ENDIAN = 0, THP_BIG_ENDIAN = 1 }; TORCH_API THPByteOrder THP_nativeByteOrder(); TORCH_API void THP_decodeInt16Buffer( int16_t* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeInt32Buffer( int32_t* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeInt64Buffer( int64_t* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeHalfBuffer( THHalf* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeFloatBuffer( float* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeDoubleBuffer( double* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeBoolBuffer( bool* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeBFloat16Buffer( at::BFloat16* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeComplexFloatBuffer( c10::complex* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_decodeComplexDoubleBuffer( c10::complex* dst, const uint8_t* src, THPByteOrder order, size_t len); TORCH_API void THP_encodeInt16Buffer( uint8_t* dst, const int16_t* src, THPByteOrder order, size_t len); TORCH_API void THP_encodeInt32Buffer( uint8_t* dst, const int32_t* src, THPByteOrder order, size_t len); TORCH_API void THP_encodeInt64Buffer( uint8_t* dst, const int64_t* src, THPByteOrder order, size_t len); TORCH_API void THP_encodeFloatBuffer( uint8_t* dst, const float* src, THPByteOrder order, size_t len); TORCH_API void THP_encodeDoubleBuffer( uint8_t* dst, const double* src, THPByteOrder order, size_t len); TORCH_API void THP_encodeComplexloatBuffer( uint8_t* dst, const c10::complex* src, THPByteOrder order, size_t len); TORCH_API void THP_encodeComplexDoubleBuffer( uint8_t* dst, const c10::complex* src, THPByteOrder order, size_t len); } // namespace utils } // namespace torch