diff options
author | slaren <slarengh@gmail.com> | 2023-05-02 16:03:00 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-02 16:03:00 +0200 |
commit | 2d099e5193d73f800b646c39e2fad08c1c1f1096 (patch) | |
tree | 7ffd7e30d86ad6eeb1be5eca2ff3f1978e269a11 /ggml.h | |
parent | f4cef87edfd1b2f8d5befd4fde54ca2e03987bea (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.h | 8 |
1 files changed, 7 insertions, 1 deletions
@@ -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 // |