ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Loss.cc
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2023 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
#include "
ir/operation/Loss.h
"
18
#include "
ir/OperationVisitor.h
"
19
20
#include <unordered_map>
21
22
namespace
onert
23
{
24
namespace
ir
25
{
26
namespace
operation
27
{
28
29
void
Loss::accept
(
OperationVisitor
&v)
const
{ v.visit(*
this
); }
30
31
Loss::Loss
(
const
OperandIndexSequence
&inputs,
const
OperandIndexSequence
&outputs)
32
:
Operation
{
OperandConstraint
::createAtLeast(2u), inputs, outputs}
33
{
34
assert(inputs.size() == 2);
35
}
36
37
}
// namespace operation
38
}
// namespace ir
39
}
// namespace onert
OperationVisitor.h
onert::ir::OperandConstraint
Definition
OperandConstraint.h:30
onert::ir::OperandIndexSequence
Definition
OperandIndexSequence.h:37
onert::ir::Operation
Definition
Operation.h:34
onert::ir::operation::Loss::Loss
Loss(const OperandIndexSequence &inputs, const OperandIndexSequence &outputs)
Definition
Loss.cc:31
onert::ir::operation::Loss::accept
void accept(OperationVisitor &v) const override
Definition
Loss.cc:29
onert
Definition
CustomKernel.cc:20
Loss.h
onert::ir::OperationVisitor
Definition
OperationVisitor.h:28
runtime
onert
core
src
ir
operation
Loss.cc
Generated by
1.9.8