ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CrossEntropy.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2024 Samsung Electronics Co., Ltd. All Rights Reserved
3
*
4
* Licensed under the Apache License, Version 2.0 (the "License");
5
* you may not use this file except in compliance with the License.
6
* You may obtain a copy of the License at
7
*
8
* http://www.apache.org/licenses/LICENSE-2.0
9
*
10
* Unless required by applicable law or agreed to in writing, software
11
* distributed under the License is distributed on an "AS IS" BASIS,
12
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
* See the License for the specific language governing permissions and
14
* limitations under the License.
15
*/
16
17
#ifndef ONERT_MICRO_TRAIN_METRICS_CROSS_ENTROPY_H
18
#define ONERT_MICRO_TRAIN_METRICS_CROSS_ENTROPY_H
19
20
#include "
OMStatus.h
"
21
22
#include <cstdint>
23
24
namespace
onert_micro
25
{
26
namespace
train
27
{
28
namespace
metrics
29
{
30
31
struct
CrossEntropy
32
{
33
// Calculate cross entropy metric between calculated and target data
34
static
float
calculateValue
(
const
uint32_t flat_size,
const
float
*calculated_data,
35
const
float
*target_data);
36
};
37
38
}
// namespace metrics
39
}
// namespace train
40
}
// namespace onert_micro
41
42
#endif
// ONERT_MICRO_TRAIN_METRICS_CROSS_ENTROPY_H
OMStatus.h
onert_micro
Definition
OMMemoryManager.h:26
onert_micro::train::metrics::CrossEntropy
Definition
CrossEntropy.h:32
onert_micro::train::metrics::CrossEntropy::calculateValue
static float calculateValue(const uint32_t flat_size, const float *calculated_data, const float *target_data)
Definition
CrossEntropy.cpp:31
onert-micro
onert-micro
include
train
metrics
CrossEntropy.h
Generated by
1.9.8