Home
last modified time | relevance | path

Searched defs:expanded_shape (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dmatrix_band_part.h31 std::vector<T> expanded_shape; in GetExpandedShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.cc460 Arrangement expanded_shape; in ComputeExpandedTensorShape() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/
D_constexpr_utils.py714 def expanded_shape(shape, expand_size): function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dgrad_nn_ops.cc275 ShapeVector expanded_shape; in Calc() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/
D_checkparam.py963 def expanded_shape(ndim, axis_size, axis): function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/
Dstandard_method.py3224 expanded_shape = validator.expanded_shape variable