Searched refs:arg_shape (Results 1 – 2 of 2) sorted by relevance
163 ShapeVector arg_shape = arg_spec->shape()->shape(); in InferImplBatchNorm() local164 if (arg_shape.size() != 1) { in InferImplBatchNorm()165 MS_LOG(EXCEPTION) << "Arg " << i << " rank should be 1, but got " << arg_shape.size(); in InferImplBatchNorm()167 if ((x_shape[c_axis] != Shape::SHP_ANY) && (arg_shape[0] != x_shape[c_axis])) { in InferImplBatchNorm()169 << "]=" << x_shape[c_axis] << ", but got " << arg_shape[0]; in InferImplBatchNorm()
116 py::tuple arg_shape = py_args[i].attr("shape"); in check_bprop_out() local124 if (!grad_shape.equal(arg_shape)) { in check_bprop_out()127 << "th arg shape is: " << py::cast<py::str>(arg_shape) in check_bprop_out()