MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / SaveArrayAsNumpy

Function SaveArrayAsNumpy

include/libnpy/npy.hpp:491–509  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

489
490template<typename Scalar>
491inline void
492SaveArrayAsNumpy(const std::string &filename, bool fortran_order, unsigned int n_dims, const unsigned long shape[],
493 const Scalar* data) {
494// static_assert(has_typestring<Scalar>::value, "scalar type not understood");
495 const dtype_t dtype = dtype_map.at(std::type_index(typeid(Scalar)));
496
497 std::ofstream stream(filename, std::ofstream::binary);
498 if (!stream) {
499 throw std::runtime_error("io error: failed to open a file.");
500 }
501
502 std::vector <ndarray_len_t> shape_v(shape, shape + n_dims);
503 header_t header{dtype, fortran_order, shape_v};
504 write_header(stream, header);
505
506 auto size = static_cast<size_t>(comp_size(shape_v));
507
508 stream.write(reinterpret_cast<const char *>(data), sizeof(Scalar) * size);
509}
510
511template<typename Scalar>
512inline void

Callers

nothing calls this directly

Calls 3

write_headerFunction · 0.85
comp_sizeFunction · 0.85
dataMethod · 0.45

Tested by

no test coverage detected