Searched refs:c_axis (Results 1 – 2 of 2) sorted by relevance
155 size_t c_axis = 1; in InferImplBatchNorm() local157 c_axis = 3; in InferImplBatchNorm()167 if ((x_shape[c_axis] != Shape::SHP_ANY) && (arg_shape[0] != x_shape[c_axis])) { in InferImplBatchNorm()168 MS_EXCEPTION(ValueError) << "Arg " << i << " shape[0] should equal to x_shape[" << c_axis in InferImplBatchNorm()169 << "]=" << x_shape[c_axis] << ", but got " << arg_shape[0]; in InferImplBatchNorm()264 uint64_t c_axis = 1; in InferImplConv2D() local269 c_axis = 3; in InferImplConv2D()274 if ((x_shape[c_axis] != Shape::SHP_ANY) && (w_shape[c_axis] != Shape::SHP_ANY) && in InferImplConv2D()275 ((x_shape[c_axis] / group) != w_shape[c_axis])) { in InferImplConv2D()276 …MS_LOG(EXCEPTION) << "x_shape[C_in] / group must equal to w_shape[C_in] = " << w_shape[c_axis] << … in InferImplConv2D()[all …]
172 uint64_t c_axis = 1; in Conv2dInferShape() local177 c_axis = 3; in Conv2dInferShape()182 if ((x_shape[c_axis] != Shape::SHP_ANY) && (w_shape[c_axis] != Shape::SHP_ANY) && in Conv2dInferShape()183 ((x_shape[c_axis] / group) != w_shape[c_axis])) { in Conv2dInferShape()184 …MS_LOG(EXCEPTION) << "x_shape[C_in] / group must equal to w_shape[C_in] = " << w_shape[c_axis] << … in Conv2dInferShape()185 << (x_shape[c_axis] / group); in Conv2dInferShape()