gguf : add gguf_get_tensor_name()

ggerganov · Jul 26, 2023 · 480a067 · 480a067
1 parent f72dfbf
commit 480a067
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/ggml.c b/ggml.c
@@ -18741,6 +18741,10 @@ size_t gguf_get_tensor_offset(struct gguf_context * ctx, int i) {
     return ctx->infos[i].offset;
 }
 
+char * gguf_get_tensor_name(struct gguf_context * ctx, int i) {
+    return ctx->infos[i].name.data;
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 
 int ggml_cpu_has_avx(void) {

diff --git a/ggml.h b/ggml.h
@@ -1653,6 +1653,7 @@ extern "C" {
 
     GGML_API int    gguf_get_n_tensors    (struct gguf_context * ctx);
     GGML_API size_t gguf_get_tensor_offset(struct gguf_context * ctx, int i);
+    GGML_API char * gguf_get_tensor_name  (struct gguf_context * ctx, int i);
 
     //
     // system info