Neo Zhang
55edb2de44
* support OP cross_entropy_loss, cross_entropy_loss_back * correct format issue
8 lines
217 B
C++
8 lines
217 B
C++
#pragma once
|
|
|
|
#include "common.hpp"
|
|
|
|
void ggml_sycl_cross_entropy_loss(ggml_backend_sycl_context & ctx, ggml_tensor * dst);
|
|
|
|
void ggml_sycl_cross_entropy_loss_back(ggml_backend_sycl_context & ctx, ggml_tensor * dst);
|