Searched refs:LeakyRelu (Results 1 – 16 of 16) sorted by relevance
/third_party/mindspore/mindspore/core/ops/ |
D | leaky_relu.h | 32 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>;
|
D | leaky_relu.cc | 41 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/ |
D | nonlinear_fuc_ops_declare.cc | 159 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))
|
D | nonlinear_fuc_ops_declare.h | 95 DECLARE_OP_ADAPTER(LeakyRelu) 96 DECLARE_OP_USE_OUTPUT(LeakyRelu)
|
/third_party/mindspore/tests/ut/cpp/ops/ |
D | test_ops_leakyrelu.cc | 35 auto leakyrelu = std::make_shared<LeakyRelu>(); in TEST_F()
|
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
D | activation_mapper.cc | 39 {mindspore::LEAKY_RELU, std::make_shared<ops::LeakyRelu>()}}; in Mapper()
|
/third_party/mindspore/mindspore/lite/src/ops/ |
D | ops_func_declare.h | 336 FUNC_MSOP2SCHEMAOP_DECLARE(LeakyRelu)
|
D | ops_def.cc | 95 OP_TYPE(LeakyRelu) 612 OP_SCHEMA_DEF(LeakyRelu) 614 OP_SCHEMA_DEF_END(LeakyRelu)
|
D | ops_utils.cc | 345 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LeakyRelu>>(node); in LeakyReluPrimitiveCreator()
|
/third_party/mindspore/mindspore/lite/schema/ |
D | ops.fbs | 96 LeakyRelu, 612 table LeakyRelu {
|
D | model_generated.h | 692 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…
|
D | ops_generated.h | 239 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/ |
D | activation.cl | 20 __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/ |
D | common_infer.c | 463 REG_INFER(LeakyRelu, PrimType_LeakyRelu, CommonInferShape)
|
/third_party/mindspore/mindspore/lite/schema/inner/ |
D | model_generated.h | 841 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…
|
D | ops_generated.h | 314 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 …]
|