Home
last modified time | relevance | path

Searched refs:update_rank (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dscatter_nd_fp32.cc71 auto update_rank = static_cast<int>(update->shape().size()); in ReSize() local
73 …if (update_rank != static_cast<int>(indices->shape().size() - 1 + shape_rank - indice_unit_rank)) { in ReSize()
95 for (int i = indices_shape.size() - 1; i < update_rank; i++) { in ReSize()
Dscatter_nd_update_fp32.cc69 int update_rank = static_cast<int>(update->shape().size()); in ReSize() local
73 for (int i = indices_shape.size() - 1; i < update_rank; i++) { in ReSize()