Searched defs:use_axis (Results 1 – 3 of 3) sorted by relevance
38 void FullConnection::set_use_axis(const bool use_axis) { (void)this->AddAttr(kUseAxis, MakeValue(us… in set_use_axis()54 void FullConnection::Init(const bool has_bias, const int64_t axis, const bool use_axis, in Init()82 auto use_axis = GetValue<bool>(primitive->GetAttr(kUseAxis)); in FullConnectionInfer() local
37 bool use_axis = false; in TEST_F() local67 bool use_axis = false; in TEST_F() local98 bool use_axis = true; in TEST_F() local
1751 PrimitivePtr MindIR_FullConnection_CreatePrimitive(bool has_bias, bool use_axis, int64_t axis, in MindIR_FullConnection_CreatePrimitive()1808 void MindIR_FullConnection_SetUseAxis(PrimitivePtr *primitive, bool use_axis) { in MindIR_FullConnection_SetUseAxis()