| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | gather_cpu_kernel.cc | 54 const auto *indices_data = reinterpret_cast<int32_t *>(inputs[1]->addr); in Launch() local
69 ParallelRun(input_tensor, indices_data, output_addr, max_thread_num); in Launch()
74 void GatherV2CPUKernel<T>::ParallelRun(const int8_t *input_addr, const int *indices_data, int8_t *o… in ParallelRun() argument
100 …auto block = [this, in, indices_data, count, inner_size, limit, indices_element_size, out, thread_… in ParallelRun()
101 …int ret = Gather(in, count, inner_size, limit, indices_data, indices_element_size, out, sizeof(T)); in ParallelRun()
|
| D | gather_cpu_kernel.h | 40 …void ParallelRun(const int8_t *input_addr, const int *indices_data, int8_t *output_addr, int threa…
|
| /third_party/boost/libs/compute/test/ |
| D | test_gather.cpp | 29 int indices_data[] = { 0, 4, 1, 3, 2 }; in BOOST_AUTO_TEST_CASE() local
30 compute::vector<int> indices(indices_data, indices_data + 5, queue); in BOOST_AUTO_TEST_CASE()
|
| D | test_permutation_iterator.cpp | 86 int indices_data[] = { 3, 2, 1, 0 }; in BOOST_AUTO_TEST_CASE() local
89 boost::compute::vector<int> indices(indices_data, indices_data + 4, queue); in BOOST_AUTO_TEST_CASE()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | one_hot_base.cc | 105 auto indices_data = static_cast<int *>(in_tensors_.at(0)->data()); in OneHotImpl() local
106 if (indices_data == nullptr) { in OneHotImpl()
121 …auto ret = OneHotToFp32(indices_data, on_value_, off_value_, reinterpret_cast<float *>(output_data… in OneHotImpl()
126 …auto ret = OneHotToFp16(indices_data, on_value_, off_value_, reinterpret_cast<float16_t *>(output_… in OneHotImpl()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | one_hot_fp32_test.cc | 29 …void Prepare(const std::vector<int> &indices_shape, int *indices_data, int *depth, float *off_on_v…
57 void TestOneHotFp32::Prepare(const std::vector<int> &indices_shape, int *indices_data, int *depth, … in Prepare() argument
62 indices_tensor_.set_data(indices_data); in Prepare()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info.cc | 211 int *indices_data = reinterpret_cast<int *>(indices()->addr); in ComputeMean() local
226 indices_data[j] -= SizeToInt(offset); in ComputeMean()
231 …Util::ReduceSparseGradient(grad_data, indices_data, indices_size, segment_size, first_dim_size, ou… in ComputeMean()
|
| /third_party/vk-gl-cts/external/openglcts/modules/gl/ |
| D | gl4cVertexAttrib64BitTest.cpp | 2673 std::vector<GLuint> indices_data; in testInit() local
2674 indices_data.resize(m_n_vertices); in testInit()
2677 indices_data[i] = i; in testInit()
2697 …gl.bufferData(GL_ELEMENT_ARRAY_BUFFER, m_n_vertices * sizeof(GLuint), &indices_data[0], GL_STATIC_… in testInit()
|
| D | gl4cShadingLanguage420PackTests.cpp | 16496 static const GLuint indices_data[4] = { 2, 1, 0, 3 }; in prepareUniforms() local
16508 program.uniform("indices", Utils::UINT, 1 /* n_cols */, 4 /* n_rows */, indices_data); in prepareUniforms()
|