Home
last modified time | relevance | path

Searched defs:arg_shape (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dprimitive_py.cc229 py::tuple arg_shape = py_args[i].attr("shape"); in check_bprop_input_grads() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dop_utils.cc642 auto arg_shape = arg->GetShape()->GetShapeVector(); in GetShapeValue() local