MCPcopy Create free account
hub / github.com/Tiiny-AI/PowerInfer / assert_shape_4d

Function assert_shape_4d

common/train.cpp:203–209  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

201}
202
203void assert_shape_4d(struct ggml_tensor * tensor, int64_t ne0, int64_t ne1, int64_t ne2, int64_t ne3) {
204 GGML_ASSERT(tensor->n_dims == 4);
205 GGML_ASSERT(tensor->ne[0] == ne0);
206 GGML_ASSERT(tensor->ne[1] == ne1);
207 GGML_ASSERT(tensor->ne[2] == ne2);
208 GGML_ASSERT(tensor->ne[3] == ne3);
209}
210
211int64_t get_example_targets_batch(
212 struct llama_context * lctx,

Callers 3

forward_batchFunction · 0.85
llama_build_train_graphsFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected