Searched refs:output_shape (Results 1 – 3 of 3) sorted by relevance
168 std::vector<size_t> output_shape; in save_npy() local171 output_shape.push_back(output_attr->dims[i]); in save_npy()174 npy_save<float>(output_path, output_data, output_shape); in save_npy()
171 std::vector<size_t> output_shape; in save_npy() local175 output_shape.push_back(output_attr->dims[i]); in save_npy()178 npy_save<T>(output_path, output_data, output_shape); in save_npy()