Home
last modified time | relevance | path

Searched refs:LeakyRelu (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dleaky_relu.h32 class MS_CORE_API LeakyRelu : public PrimitiveC {
35 LeakyRelu() : PrimitiveC(kNameLeakyRelu) {} in LeakyRelu() function
37 ~LeakyRelu() = default;
38 MS_DECLARE_PARENT(LeakyRelu, PrimitiveC);
51 using PrimLeakyReluPtr = std::shared_ptr<LeakyRelu>;
Dleaky_relu.cc41 void LeakyRelu::Init(const float negative_slope) { this->set_negative_slope(negative_slope); } in Init()
43 void LeakyRelu::set_negative_slope(const float negative_slope) { in set_negative_slope()
46 float LeakyRelu::get_negative_slope() const { return GetValue<float>(GetAttr(kNegativeSlope)); } in get_negative_slope()
53 REGISTER_PRIMITIVE_C(kNameLeakyRelu, LeakyRelu);
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dnonlinear_fuc_ops_declare.cc159 INPUT_MAP(LeakyRelu) = {{1, INPUT_DESC(x)}};
160 ATTR_MAP(LeakyRelu) = {{"alpha", ATTR_DESC(negative_slope, AnyTraits<float>())}};
161 OUTPUT_MAP(LeakyRelu) = {{0, OUTPUT_DESC(y)}};
162 REG_ADPT_DESC(LeakyRelu, prim::kPrimLeakyRelu->name(), ADPT_DESC(LeakyRelu))
Dnonlinear_fuc_ops_declare.h95 DECLARE_OP_ADAPTER(LeakyRelu)
96 DECLARE_OP_USE_OUTPUT(LeakyRelu)
/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_leakyrelu.cc35 auto leakyrelu = std::make_shared<LeakyRelu>(); in TEST_F()
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dactivation_mapper.cc39 {mindspore::LEAKY_RELU, std::make_shared<ops::LeakyRelu>()}}; in Mapper()
/third_party/mindspore/mindspore/lite/src/ops/
Dops_func_declare.h336 FUNC_MSOP2SCHEMAOP_DECLARE(LeakyRelu)
Dops_def.cc95 OP_TYPE(LeakyRelu)
612 OP_SCHEMA_DEF(LeakyRelu)
614 OP_SCHEMA_DEF_END(LeakyRelu)
Dops_utils.cc345 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LeakyRelu>>(node); in LeakyReluPrimitiveCreator()
/third_party/mindspore/mindspore/lite/schema/
Dops.fbs96 LeakyRelu,
612 table LeakyRelu {
Dmodel_generated.h692 const mindspore::schema::LeakyRelu *value_as_LeakyRelu() const { in value_as_LeakyRelu()
693 …:schema::PrimitiveType_LeakyRelu ? static_cast<const mindspore::schema::LeakyRelu *>(value()) : nu… in value_as_LeakyRelu()
1376 template<> inline const mindspore::schema::LeakyRelu *Primitive::value_as<mindspore::schema::LeakyR…
Dops_generated.h239 struct LeakyRelu;
1545 template<> struct PrimitiveTypeTraits<mindspore::schema::LeakyRelu> {
6315 struct LeakyRelu FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
6331 typedef LeakyRelu Table;
6335 fbb_.AddElement<float>(LeakyRelu::VT_NEGATIVE_SLOPE, negative_slope, 0.0f);
6341 flatbuffers::Offset<LeakyRelu> Finish() {
6343 auto o = flatbuffers::Offset<LeakyRelu>(end);
6348 inline flatbuffers::Offset<LeakyRelu> CreateLeakyRelu(
12962 auto ptr = reinterpret_cast<const mindspore::schema::LeakyRelu *>(obj);
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/cl/
Dactivation.cl20 __kernel void LeakyRelu(__read_only image2d_t input, __write_only image2d_t output, const int2 img_…
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dcommon_infer.c463 REG_INFER(LeakyRelu, PrimType_LeakyRelu, CommonInferShape)
/third_party/mindspore/mindspore/lite/schema/inner/
Dmodel_generated.h841 const mindspore::schema::LeakyRelu *value_as_LeakyRelu() const { in value_as_LeakyRelu()
842 …:schema::PrimitiveType_LeakyRelu ? static_cast<const mindspore::schema::LeakyRelu *>(value()) : nu… in value_as_LeakyRelu()
1531 template<> inline const mindspore::schema::LeakyRelu *Primitive::value_as<mindspore::schema::LeakyR…
Dops_generated.h314 struct LeakyRelu;
2145 template<> struct PrimitiveTypeTraits<mindspore::schema::LeakyRelu> {
10181 typedef LeakyRelu TableType;
10185 struct LeakyRelu FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
10207 …static flatbuffers::Offset<LeakyRelu> Pack(flatbuffers::FlatBufferBuilder &_fbb, const LeakyReluT*…
10211 typedef LeakyRelu Table;
10215 fbb_.AddElement<float>(LeakyRelu::VT_NEGATIVE_SLOPE, negative_slope, 0.0f);
10221 flatbuffers::Offset<LeakyRelu> Finish() {
10223 auto o = flatbuffers::Offset<LeakyRelu>(end);
10228 inline flatbuffers::Offset<LeakyRelu> CreateLeakyRelu(
[all …]