Searched defs:use_axis (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/ |
| D | full_connection.cc | 43 void FullConnection::set_use_axis(const bool use_axis) { (void)this->AddAttr(kUseAxis, api::MakeVal… in set_use_axis() 59 void FullConnection::Init(const bool has_bias, const int64_t axis, const bool use_axis, in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir_2.cc | 212 …hannel, int64_t out_channel, EltwiseMode eltwise_mode, bool has_bias, bool use_axis, int64_t axis,… in MindIR_GenOP_CreatePrimitive() 799 void MindIR_GenOP_SetUseAxis(PrimitivePtr *primitive, bool use_axis) { in MindIR_GenOP_SetUseAxis()
|
| D | mindir.cc | 1854 PrimitivePtr MindIR_FullConnection_CreatePrimitive(bool has_bias, bool use_axis, int64_t axis, in MindIR_FullConnection_CreatePrimitive() 1911 void MindIR_FullConnection_SetUseAxis(PrimitivePtr *primitive, bool use_axis) { in MindIR_FullConnection_SetUseAxis()
|