Home
last modified time | relevance | path

Searched refs:dim3 (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dstrided_slice_fp32.c82 int32_t dim0, dim1, dim2, dim3, dim4, dim5, dim6, dim7; in DoStridedSliceIntFp64Bool() local
86 for (dim3 = begins[3]; LoopContinue(strides[3], dim3, ends[3]); dim3 += strides[3]) { in DoStridedSliceIntFp64Bool()
92dim3 * dim_offset[3] + dim4 * dim_offset[4] + dim5 * dim_offset[5] + in DoStridedSliceIntFp64Bool()
144 int32_t dim0, dim1, dim2, dim3, dim4, dim5, dim6, dim7; in DoStridedSlice() local
148 for (dim3 = begins[3]; LoopContinue(strides[3], dim3, ends[3]); dim3 += strides[3]) { in DoStridedSlice()
154dim3 * dim_offset[3] + dim4 * dim_offset[4] + dim5 * dim_offset[5] + in DoStridedSlice()
/third_party/boost/libs/fiber/examples/hip/
Dmultiple_streams.cpp65 …hipLaunchKernel( vector_add, dim3(size / 256), dim3(256), 0, stream0, dev_a0, dev_b0, dev_c0, size… in main()
66 …hipLaunchKernel( vector_add, dim3(size / 256), dim3(256), 0, stream1, dev_a1, dev_b1, dev_c1, size… in main()
Dsingle_stream.cpp58 … hipLaunchKernel( vector_add, dim3(size / 256), dim3(256), 0, stream, dev_a, dev_b, dev_c, size); in main()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dcommon_func.c19 int Offset(const int *shape, const int dim0, const int dim1, const int dim2, const int dim3) { in Offset() argument
20 return ((dim0 * shape[1] + dim1) * shape[2] + dim2) * shape[3] + dim3; in Offset()
Dcommon_func.h31 int Offset(const int *shape, const int dim0, const int dim1, const int dim2, const int dim3);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dslice_base.c137 for (int32_t dim3 = param->begin_[3]; dim3 < param->end_[3]; ++dim3) { in DoSliceNoParallel() local
138 int in_offset3 = in_offset2 + dim3 * in_stride[3]; in DoSliceNoParallel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/pme/
Dpme_energy_impl.cuh27dim3 thread_PME, int PME_Nin, int PME_Nfft, int PME_Nall, const cufftHandle &PME_plan_r2c,
Dpme_energy_update_impl.cuh27 … float *d_correction_ene, dim3 thread_PME, int PME_Nin, int PME_Nfft, int PME_Nall,
Dpme_energy_update_impl.cu44 … float *d_correction_ene, dim3 thread_PME, int PME_Nin, int PME_Nfft, int PME_Nall, in PMEEnergyUpdate()
Dpme_energy_impl.cu43dim3 thread_PME, int PME_Nin, int PME_Nfft, int PME_Nall, const cufftHandle &PME_plan_r2c, in PMEEnergy()
Dpme_common.cuh33 static dim3 thread_PME;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dbias_add_grad_impl.cu159 dim3 grid_size(block_num_x, block_num_y, 1); in CalBiasAddGradNHWC()
160 dim3 block_size(kWarpSize*kWarpSize); in CalBiasAddGradNHWC()
/third_party/boost/libs/fiber/doc/
Dhip.qbk60 … hipLaunchKernel(kernel, dim3(size / 256), dim3(256), 0, stream, size, dev_a, dev_b, dev_c);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel.cc68 size_t dim3) { in CalcOffset() argument
69 …fset = dim0 * shape[1] * shape[2] * shape[3] + dim1 * shape[2] * shape[3] + dim2 * shape[3] + dim3; in CalcOffset()
Dcpu_kernel.h146 …t CalcOffset(const std::vector<size_t> &shape, size_t dim0, size_t dim1, size_t dim2, size_t dim3);
/third_party/ffmpeg/compat/cuda/
Dcuda_runtime.h67 typedef struct uint3 dim3; typedef
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/pme/
Dpme_energy_kernel.h259 dim3 _thread_PME;
Dpme_energy_update_kernel.h292 dim3 _thread_PME;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/
Dcommon_sponge.cuh35 static dim3 thread_LJ(8, 32);
/third_party/skia/third_party/externals/angle2/src/tests/gl_tests/
DGLSLTest.cpp10975 for (int dim3 = 0; dim3 < 4; ++dim3) in TEST_P() local
10977 int textureUnit = (dim1 * 3 + dim2) * 4 + dim3; in TEST_P()
10979 glBindTexture(GL_TEXTURE_2D, textures[dim1][dim2][dim3]); in TEST_P()
10986 uniformName << "smplr[" << dim1 << "][" << dim2 << "][" << dim3 << "]"; in TEST_P()
11152 for (int dim3 = 0; dim3 < 4; ++dim3) in TEST_P() local
11154 int textureUnit = (dim1 * 3 + dim2) * 4 + dim3; in TEST_P()
11156 glBindTexture(GL_TEXTURE_2D, textures[dim1][dim2][dim3]); in TEST_P()
11163 uniformName << "u[" << dim1 << "].nested[" << dim2 << "].smplr[" << dim3 << "]"; in TEST_P()