Home
last modified time | relevance | path

Searched defs:input_nnz (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_fill_empty_rows.cc105 const int64_t input_nnz = indice_shape[0]; in SparseFillEmptyRowsInferShape() local
139 const int64_t input_nnz = indice_shape[0]; in SparseFillEmptyRowsFrontendInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse_grad/
Dsparse_slice_grad_gpu_kernel.cc74 const int64_t input_nnz = input_indices_shape[0]; in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_slice_grad.cc92 const int64_t input_nnz = indices_shape->GetDimSize(0); in GradCompute() local