Home
last modified time | relevance | path

Searched defs:matrix_index (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dlu_unpack.cc49 …ckCpuKernel::LuUnpack(CpuKernelContext &ctx, T_pivots *Lu_pivots_working_ptr, int64_t matrix_index, in LuUnpack()
160 for (int64_t matrix_index = 0; matrix_index < batch_num; matrix_index++) { in LuUnpackCompute() local
175 for (int64_t matrix_index = start; matrix_index < end; matrix_index++) { in LuUnpackCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.cc63 int64_t matrix_index = this->in_tensor_map_.GetDimByIdx(i); in MergePartialToFullForReshapeHasMultiDynamicAxis() local
83 int64_t matrix_index = this->out_tensor_map_.GetDimByIdx(i); in SegmentFullShapeToPartial() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dlu_unpack_cpu_kernel.cc91 … int64_t Lu_pivots_dim, T_pivots *const Lu_pivots_working_ptr, int64_t matrix_index, in LuUnpack()
209 for (int64_t matrix_index = 0; matrix_index < batch_num; matrix_index++) { in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dreduce_impl.cu385 const int matrix_index = gid / num_cols; in ColumnReduceSimpleKernel() local