| /third_party/mindspore/tests/ut/cpp/kernel/cpu/ |
| D | sparse_apply_ftrl_cpu_kernel_test.cc | 59 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() argument
62 workspace_.push_back(CreateKernelAddress(new_indices.data())); in CreateWorkspaceAddress()
92 std::vector<int64_t> new_indices(3); in TEST_F() local
95 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
119 std::vector<int64_t> new_indices(3); in TEST_F() local
122 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
150 std::vector<int64_t> new_indices(3); in TEST_F() local
153 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
|
| D | sparse_apply_proximal_adagrad_cpu_kernel_test.cc | 57 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() argument
60 workspace_.push_back(CreateKernelAddress(new_indices.data())); in CreateWorkspaceAddress()
91 std::vector<int64_t> new_indices(3); in TEST_F() local
94 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
117 std::vector<int64_t> new_indices(3); in TEST_F() local
120 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
147 std::vector<int64_t> new_indices(3); in TEST_F() local
150 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
|
| D | sparse_apply_lazy_adam_cpu_kernel_test.cc | 61 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() argument
64 workspace_.push_back(CreateKernelAddress(new_indices.data())); in CreateWorkspaceAddress()
100 std::vector<int64_t> new_indices(3); in TEST_F() local
103 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
127 std::vector<int64_t> new_indices(3); in TEST_F() local
130 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
158 std::vector<int64_t> new_indices(3); in TEST_F() local
161 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
|
| D | sparse_apply_adam_cpu_kernel_test.cc | 61 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() argument
65 workspace_.push_back(CreateKernelAddress(new_indices.data())); in CreateWorkspaceAddress()
102 std::vector<int64_t> new_indices(3); in TEST_F() local
106 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
130 std::vector<int64_t> new_indices(3); in TEST_F() local
134 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
162 std::vector<int64_t> new_indices(3); in TEST_F() local
166 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 137 auto new_indices = reinterpret_cast<T *>(workspace[1]->addr); in LaunchKernel() local
141 SparseGradient<T> unique_sparse_grad({new_grad, new_indices, indices_size_}); in LaunchKernel()
|
| D | sparse_apply_ftrl_cpu_kernel.cc | 148 auto *new_indices = reinterpret_cast<T *>(workspace[1]->addr); in LaunchKernel() local
152 SparseGradient<T> unique_sparse_grad({new_grad, new_indices, indices_size_}); in LaunchKernel()
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 140 auto *new_indices = reinterpret_cast<T *>(workspace[1]->addr); in LaunchKernel() local
144 SparseGradient<T> unique_sparse_grad({new_grad, new_indices, indices_size_}); in LaunchKernel()
|
| D | sparse_apply_adam_cpu_kernel.cc | 160 auto *new_indices = reinterpret_cast<T *>(workspace[1]->addr); in LaunchKernel() local
165 SparseGradient<T> unique_sparse_grad({new_grad, new_indices, indices_size_}); in LaunchKernel()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 276 new_indices = sub_op(indices, offset)
277 indices_size = size_op(new_indices)
281 new_indices = reshape_op(new_indices, new_indices_shape_changed)
288 return RowTensor(new_indices, actual_dout, x_shp), zeros_like(indices), zeros_like(offset)
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info.cc | 191 std::vector<int> new_indices(indices_size); in ComputeMean() local
192 …mindspore::kernel::SparseGradient<int> unique_sparse_grad({new_grad.data(), new_indices.data(), in… in ComputeMean()
|
| D | worker.cc | 799 std::vector<int> new_indices(indices_size); in SparsePartitioner() local
800 …mindspore::kernel::SparseGradient<int> unique_sparse_grad({new_grad.data(), new_indices.data(), in… in SparsePartitioner()
|
| /third_party/mesa3d/src/amd/vulkan/ |
| D | radv_pipeline_rt.c | 1531 …nir_ssa_def *new_indices[4] = {nir_ssa_undef(b, 1, 32), nir_ssa_undef(b, 1, 32), nir_ssa_undef(b… in nir_sort_hit_pair() local
1534 new_indices[chan_2] = nir_channel(b, ssa_indices, chan_1); in nir_sort_hit_pair()
1535 new_indices[chan_1] = nir_channel(b, ssa_indices, chan_2); in nir_sort_hit_pair()
1537 nir_store_var(b, var_indices, nir_vec(b, new_indices, 4), (1u << chan_1) | (1u << chan_2)); in nir_sort_hit_pair()
|