Searched refs:use_axis_ (Results 1 – 6 of 6) sorted by relevance
24 if (param->use_axis_ && (param->axis_ < 1 || param->axis_ > (int)(input0->shape_size_))) { in FullConnectionInferPreJudge()50 if (param->use_axis_) { in FullConnectionInferShape()71 if (param->use_axis_) { in FullConnectionInferShape()
44 param->use_axis_ = false; in TEST_F()75 param->use_axis_ = false; in TEST_F()106 param->use_axis_ = false; in TEST_F()
50 param->use_axis_ = value->use_axis(); in PopulateFullconnectionParameter()
50 matmul_param->use_axis_ = full_connection_prim->useAxis(); in PopulateFullconnectionParameter()
63 bool use_axis_; member
82 …mat_mul_parameter.b_const_, mat_mul_parameter.act_type_, mat_mul_parameter.use_axis_, mat_mul_para… in CodeStruct()