Searched defs:new_indices (Results 1 – 11 of 11) sorted by relevance
| /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()
92 std::vector<int64_t> new_indices(3); in TEST_F() local
119 std::vector<int64_t> new_indices(3); in TEST_F() local
150 std::vector<int64_t> new_indices(3); in TEST_F() local
|
| 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()
91 std::vector<int64_t> new_indices(3); in TEST_F() local
117 std::vector<int64_t> new_indices(3); in TEST_F() local
147 std::vector<int64_t> new_indices(3); in TEST_F() local
|
| 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()
100 std::vector<int64_t> new_indices(3); in TEST_F() local
127 std::vector<int64_t> new_indices(3); in TEST_F() local
158 std::vector<int64_t> new_indices(3); in TEST_F() local
|
| D | sparse_apply_adam_cpu_kernel_test.cc | 61 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress()
102 std::vector<int64_t> new_indices(3); in TEST_F() local
130 std::vector<int64_t> new_indices(3); in TEST_F() local
162 std::vector<int64_t> new_indices(3); in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_ftrl_cpu_kernel.cc | 148 auto *new_indices = reinterpret_cast<T *>(workspace[1]->addr); in LaunchKernel() local
|
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 137 auto new_indices = reinterpret_cast<T *>(workspace[1]->addr); in LaunchKernel() local
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 140 auto *new_indices = reinterpret_cast<T *>(workspace[1]->addr); in LaunchKernel() local
|
| D | sparse_apply_adam_cpu_kernel.cc | 160 auto *new_indices = reinterpret_cast<T *>(workspace[1]->addr); in LaunchKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info.cc | 191 std::vector<int> new_indices(indices_size); in ComputeMean() local
|
| D | worker.cc | 799 std::vector<int> new_indices(indices_size); in SparsePartitioner() local
|
| /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
|