From 2d099e5193d73f800b646c39e2fad08c1c1f1096 Mon Sep 17 00:00:00 2001 From: slaren Date: Tue, 2 May 2023 16:03:00 +0200 Subject: ggml: add names to tensors (#1268) * ggml: add names to tensors * minor improvements to dot file formatting --- ggml.h | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'ggml.h') 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 // -- cgit v1.2.3