Searched defs:ComputeOffset (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | matrix_diag_part_v3_impl.cu | 21 __device__ inline int64_t ComputeOffset(const int64_t diag_idx, const int64_t num_rows, const int64… in ComputeOffset() function
|
| D | matrix_diag_v3_impl.cu | 20 __device__ inline int ComputeOffset(int diag_idx, int num_rows, int num_cols, int max_diag_len, in ComputeOffset() function
|
| D | reduce_impl.cu | 197 struct ComputeOffset { struct
198 __host__ __device__ explicit ComputeOffset(const int &cols) : cols_(cols) {} in ComputeOffset() function
200 __host__ __device__ int operator()(const int &x) const { return cols_ * x; } in operator ()()
202 int cols_;
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | arithmetic_base.cc | 419 void ArithmeticBaseCPUKernel::ComputeOffset(int task_id) { in ComputeOffset() function in mindspore::kernel::ArithmeticBaseCPUKernel
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | scatter_nd_cpu_kernel.cc | 43 void ComputeOffset(ScatterNdCpuKernelMod *content, const ComputeParams<S, T> *params, size_t num_un… in ComputeOffset() function
|
| /third_party/skia/third_party/externals/swiftshader/src/Renderer/ |
| D | Blitter.cpp | 1144 Int Blitter::ComputeOffset(Int &x, Int &y, Int &pitchB, int bytes, bool quadLayout) in ComputeOffset() function in sw::Blitter
|
| /third_party/skia/third_party/externals/swiftshader/src/Device/ |
| D | Blitter.cpp | 1478 Int Blitter::ComputeOffset(Int &x, Int &y, Int &pitchB, int bytes) in ComputeOffset() function in sw::Blitter
1483 Int Blitter::ComputeOffset(Int &x, Int &y, Int &z, Int &sliceB, Int &pitchB, int bytes) in ComputeOffset() function in sw::Blitter
|