FM / utils

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

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

Name	Size	Mode	Actions
math/	-	0755	rm
threadpool/	-	0755	rm
bench_utils.h	854	0644	edit dl rm
cast.h	1110	0644	edit dl rm
cblas.h	34183	0644	edit dl rm
conversions.h	661	0644	edit dl rm
cpuid.h	2560	0644	edit dl rm
cpu_neon.h	1578	0644	edit dl rm
eigen_utils.h	6706	0644	edit dl rm
filler.h	3987	0644	edit dl rm
fixed_divisor.h	3634	0644	edit dl rm
knobs.h	532	0644	edit dl rm
knob_patcher.h	715	0644	edit dl rm
map_utils.h	635	0644	edit dl rm
math-detail.h	1916	0644	edit dl rm
math.h	15251	0644	edit dl rm
murmur_hash3.h	909	0644	edit dl rm
proto_convert.h	263	0644	edit dl rm
proto_utils.h	13254	0644	edit dl rm
proto_wrap.h	378	0644	edit dl rm
signal_handler.h	698	0644	edit dl rm
simple_queue.h	2602	0644	edit dl rm
smart_tensor_printer.h	1402	0644	edit dl rm
string_utils.h	1206	0644	edit dl rm
zmq_helper.h	3020	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils/cast.h (1110B)

#pragma once

#include <caffe2/utils/proto_utils.h>

namespace caffe2 {

namespace cast {

inline TensorProto_DataType GetCastDataType(const ArgumentHelper& helper, std::string arg) {
  TensorProto_DataType to;
  if (helper.HasSingleArgumentOfType<string>(arg)) {
    string s = helper.GetSingleArgument<string>(arg, "float");
    std::transform(s.begin(), s.end(), s.begin(), ::toupper);
#ifndef CAFFE2_USE_LITE_PROTO
    CAFFE_ENFORCE(TensorProto_DataType_Parse(s, &to), "Unknown 'to' argument: ", s);
#else

// Manually implement in the lite proto case.
#define X(t)                         \
  if (s == #t) {                     \
    return TensorProto_DataType_##t; \
  }

X(FLOAT);
    X(INT32);
    X(BYTE);
    X(STRING);
    X(BOOL);
    X(UINT8);
    X(INT8);
    X(UINT16);
    X(INT16);
    X(INT64);
    X(FLOAT16);
    X(DOUBLE);
#undef X
    CAFFE_THROW("Unhandled type argument: ", s);

#endif
  } else {
    to = static_cast<TensorProto_DataType>(
        helper.GetSingleArgument<int>(arg, TensorProto_DataType_FLOAT));
  }
  return to;
}

};  // namespace cast

};  // namespace caffe2

#pragma once #include namespace caffe2 { namespace cast { inline TensorProto_DataType GetCastDataType(const ArgumentHelper& helper, std::string arg) { TensorProto_DataType to; if (helper.HasSingleArgumentOfType(arg)) { string s = helper.GetSingleArgument(arg, "float"); std::transform(s.begin(), s.end(), s.begin(), ::toupper); #ifndef CAFFE2_USE_LITE_PROTO CAFFE_ENFORCE(TensorProto_DataType_Parse(s, &to), "Unknown 'to' argument: ", s); #else // Manually implement in the lite proto case. #define X(t) \ if (s == #t) { \ return TensorProto_DataType_##t; \ } X(FLOAT); X(INT32); X(BYTE); X(STRING); X(BOOL); X(UINT8); X(INT8); X(UINT16); X(INT16); X(INT64); X(FLOAT16); X(DOUBLE); #undef X CAFFE_THROW("Unhandled type argument: ", s); #endif } else { to = static_cast( helper.GetSingleArgument(arg, TensorProto_DataType_FLOAT)); } return to; } }; // namespace cast }; // namespace caffe2