MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / GetOFDataTypeFromNpArray

Function GetOFDataTypeFromNpArray

oneflow/extension/python/numpy.cpp:80–83  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

78}
79
80Maybe<DataType> GetOFDataTypeFromNpArray(PyArrayObject* array) {
81 int np_array_type = PyArray_TYPE(array);
82 return NumpyTypeToOFDataType(np_array_type);
83}
84
85std::vector<size_t> OFShapeToNumpyShape(const DimVector& fixed_vec) {
86 size_t ndim = fixed_vec.size();

Callers 5

MakeLocalTensorFromDataFunction · 0.85
MakeGlobalTensorFromDataFunction · 0.85
InferScalarTypeFunction · 0.85
operator()Method · 0.85
NumpyToTensorFunction · 0.85

Calls 1

NumpyTypeToOFDataTypeFunction · 0.85

Tested by

no test coverage detected