mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-08-18 07:30:08 +02:00
ggml : add ggml_repeat_4d (llama/13824)
This commit is contained in:
committed by
Georgi Gerganov
parent
ad433403ce
commit
4dfb2c2215
@@ -935,6 +935,15 @@ extern "C" {
|
||||
struct ggml_tensor * a,
|
||||
struct ggml_tensor * b);
|
||||
|
||||
// repeat a to the specified shape
|
||||
GGML_API struct ggml_tensor * ggml_repeat_4d(
|
||||
struct ggml_context * ctx,
|
||||
struct ggml_tensor * a,
|
||||
int64_t ne0,
|
||||
int64_t ne1,
|
||||
int64_t ne2,
|
||||
int64_t ne3);
|
||||
|
||||
// sums repetitions in a into shape of b
|
||||
GGML_API struct ggml_tensor * ggml_repeat_back(
|
||||
struct ggml_context * ctx,
|
||||
|
Reference in New Issue
Block a user