| /third_party/mesa3d/src/mesa/tnl/ |
| D | t_rebase.c | 67 TYPE *tmp_indices = malloc((start + count) * sizeof(TYPE)); \
69 if (tmp_indices == NULL) { \
75 tmp_indices[start + i] = in[start + i] - min_index; \
77 return (void *)tmp_indices; \
120 void *tmp_indices = NULL; in REBASE() local
187 tmp_indices = rebase_GLuint(ptr, start, ib->count, min_index); in REBASE()
190 tmp_indices = rebase_GLushort(ptr, start, ib->count, min_index); in REBASE()
193 tmp_indices = rebase_GLubyte(ptr, start, ib->count, min_index); in REBASE()
200 if (tmp_indices == NULL) in REBASE()
204 tmp_ib.ptr = tmp_indices; in REBASE()
[all …]
|
| /third_party/mindspore/tests/ut/cpp/kernel/cpu/ |
| D | sparse_apply_ftrl_cpu_kernel_test.cc | 60 std::vector<float> &tmp_grad, std::vector<int64_t> &tmp_indices) { in CreateWorkspaceAddress() argument
64 workspace_.push_back(CreateKernelAddress(tmp_indices.data())); in CreateWorkspaceAddress()
94 std::vector<int64_t> tmp_indices(3); in TEST_F() local
95 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
121 std::vector<int64_t> tmp_indices(3); in TEST_F() local
122 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
152 std::vector<int64_t> tmp_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 | 58 std::vector<float> &tmp_grad, std::vector<int64_t> &tmp_indices) { in CreateWorkspaceAddress() argument
62 workspace_.push_back(CreateKernelAddress(tmp_indices.data())); in CreateWorkspaceAddress()
93 std::vector<int64_t> tmp_indices(3); in TEST_F() local
94 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
119 std::vector<int64_t> tmp_indices(3); in TEST_F() local
120 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
149 std::vector<int64_t> tmp_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 | 62 std::vector<float> &tmp_grad, std::vector<int64_t> &tmp_indices) { in CreateWorkspaceAddress() argument
66 workspace_.push_back(CreateKernelAddress(tmp_indices.data())); in CreateWorkspaceAddress()
102 std::vector<int64_t> tmp_indices(3); in TEST_F() local
103 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
129 std::vector<int64_t> tmp_indices(3); in TEST_F() local
130 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
160 std::vector<int64_t> tmp_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 | 62 std::vector<float> &tmp_grad, std::vector<int64_t> &tmp_indices, in CreateWorkspaceAddress() argument
67 workspace_.push_back(CreateKernelAddress(tmp_indices.data())); in CreateWorkspaceAddress()
104 std::vector<int64_t> tmp_indices(3); in TEST_F() local
106 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
132 std::vector<int64_t> tmp_indices(3); in TEST_F() local
134 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
164 std::vector<int64_t> tmp_indices(3); in TEST_F() local
166 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
|
| D | sparse_optimizer_cpu_kernel_test.cc | 51 std::vector<int> tmp_indices(6); in TEST_F() local
55 SparseGradient<int> workspace_grad({tmp_grad.data(), tmp_indices.data(), 6}); in TEST_F()
104 std::vector<int> tmp_indices(6); in TEST_F() local
107 SparseGradient<int> workspace_grad({tmp_grad.data(), tmp_indices.data(), 6}); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/ |
| D | common_utils_test.cc | 51 std::vector<int64_t> tmp_indices(6); in TEST_F() local
55 SparseGradient workspace_grad({tmp_grad.data(), tmp_indices.data(), 6}); in TEST_F()
104 std::vector<int64_t> tmp_indices(6); in TEST_F() local
107 SparseGradient workspace_grad({tmp_grad.data(), tmp_indices.data(), 6}); in TEST_F()
|