Searched refs:npy_filename (Results 1 – 1 of 1) sorted by relevance
| /external/ComputeLibrary/utils/ |
| D | Utils.h | 298 void open(const std::string &npy_filename, DataLayout file_layout = DataLayout::NCHW)
303 _fs.open(npy_filename, std::ios::in | std::ios::binary);
304 …M_COMPUTE_EXIT_ON_MSG_VAR(!_fs.good(), "Failed to load binary data from %s", npy_filename.c_str());
315 ARM_COMPUTE_ERROR_VAR("Accessing %s: %s", npy_filename.c_str(), e.what()); in catch()
586 void save_to_npy(T &tensor, const std::string &npy_filename, bool fortran_order) in save_to_npy() argument
594 fs.open(npy_filename, std::ios::out | std::ios::binary); in save_to_npy()
611 std::ofstream stream(npy_filename, std::ofstream::binary); in save_to_npy()
631 ARM_COMPUTE_ERROR_VAR("Writing %s: (%s)", npy_filename.c_str(), e.what()); in save_to_npy()
|