Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dadd_atomic_clean.cc46 ShapeVector src_shape_vec; in GetUniqReduceAxes() local
177 auto src_shape_vec = GetShape(input); in SuitableForAtomicAdd() local
203 auto src_shape_vec = GetDeviceShape(input); in SuitableForAtomicAdd() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dscatter_add_ext.cc41 auto src_shape_vec = src_shape->GetShapeVector(); in InferShape() local