aboutsummaryrefslogtreecommitdiff
path: root/ggml.h
diff options
context:
space:
mode:
authorslaren <slarengh@gmail.com>2023-05-02 16:03:00 +0200
committerGitHub <noreply@github.com>2023-05-02 16:03:00 +0200
commit2d099e5193d73f800b646c39e2fad08c1c1f1096 (patch)
tree7ffd7e30d86ad6eeb1be5eca2ff3f1978e269a11 /ggml.h
parentf4cef87edfd1b2f8d5befd4fde54ca2e03987bea (diff)
ggml: add names to tensors (#1268)
* ggml: add names to tensors * minor improvements to dot file formatting
Diffstat (limited to 'ggml.h')
-rw-r--r--ggml.h8
1 files changed, 7 insertions, 1 deletions
diff --git a/ggml.h b/ggml.h
index ef5a048..508dd69 100644
--- a/ggml.h
+++ b/ggml.h
@@ -350,7 +350,10 @@ extern "C" {
int64_t perf_time_us;
void * data;
- char padding[8];
+
+ char name[32];
+
+ char padding[8]; // TODO: remove and add padding to name?
};
// computation graph
@@ -473,6 +476,9 @@ extern "C" {
GGML_API void * ggml_get_data (const struct ggml_tensor * tensor);
GGML_API float * ggml_get_data_f32(const struct ggml_tensor * tensor);
+ GGML_API const char * ggml_get_name(const struct ggml_tensor * tensor);
+ GGML_API void ggml_set_name(struct ggml_tensor * tensor, const char * name);
+
//
// operations on tensors with backpropagation
//