| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | cumsum_fp32.c | 48 int input_offset = exclusive ? 0 : 1; in Cumsum() local
93 int input_offset = exclusive ? 0 : 1; in CumsumReverse() local
139 int input_offset = exclusive ? 0 : 1; in CumsumInt() local
183 int input_offset = exclusive ? 0 : 1; in CumsumReverseInt() local
|
| D | space_to_batch_fp32.c | 55 …int input_offset = in_b * param->in_stride_[0] + in_h * param->in_stride_[1] + in_w * param->in_st… in DoSpaceToBatch() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/ |
| D | reduce_grad.c | 35 size_t input_offset = GetInputOffset(input_num_dims, input_dims, input_iter); in ReduceMeanByAxes() local
85 size_t input_offset = GetInputOffset(num_dims, input_dims, input_iter); in ReduceSumByAxes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/expanders/ |
| D | scale_fusion.cc | 47 const auto &input_offset = inputs[kOffsetIdx]; in Expand() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/view/ |
| D | test_diagonal.cc | 35 int64_t input_offset = 1; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | argmax_impl.cu | 25 size_t input_offset = x * bound * inner_size + 0 * inner_size + y; in Argmax() local
|
| D | argmin_impl.cu | 25 size_t input_offset = x * bound * inner_size + 0 * inner_size + y; in Argmin() local
|
| D | triplet_margin_loss_impl.cu | 67 size_t input_offset = x * bound * inner_size + i * inner_size + y; in PairwiseDistance() local
131 size_t input_offset = x * bound * inner_size + i * inner_size + y; in PairwiseDistance() local
157 size_t input_offset = x * bound * inner_size + i * inner_size + y; in PairwiseDistance() local
198 size_t input_offset = x * bound * inner_size + i * inner_size + y; in PairwiseDistance() local
|
| D | deformable_offsets_grad_impl.cu | 29 … T *input_x, T *input_offset, T *output_grad_x, T *output_grad_offset) { in DeformableOffsetGradKernel()
141 … T *input_offset, T *output_grad_x, T *output_grad_offset) { in DeformableOffsetGradNHWCKernel()
207 … T *input_offset, T *output_grad_x, T *output_grad_offset) { in DeformableOffsetGradNCHWKernel()
273 … T *input_offset, T *output_grad_x, T *output_grad_offset, const uint device_id, in ApplyDeformableOffsetGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | col2im.cc | 180 void Col2imCpuKernel::InnerCompute(int64_t c_col, int64_t input_offset, int64_t output_offset, T *i… in InnerCompute()
240 int64_t input_offset = batch_input_size * elt; in Col2imCompute() local
|
| D | random_shuffle.cc | 92 auto input_offset = i * block_size; in RandomShuffleCompute() local
|
| D | deformable_offsets_grad.cc | 160 … const InputXStride &input_x_stride, const T *input_x, const T *input_offset, in DeformableOffsetGradKernel()
272 … const T *input_offset, const T *input_grad, T *output_grad_x, in DoComputeNHWC()
327 … const T *input_offset, const T *input_grad, T *output_grad_x, in DoComputeNCHW()
417 const T *input_offset = reinterpret_cast<T *>(ctx.Input(kOffsetIndex)->GetData()); in DeformableOffsetsGradTask() local
|
| D | fractional_max_pool_grad.cc | 146 int input_offset = in_index * input_size[3] + d; in DoCompute() local
190 int input_offset = in_index * input_size[3] + d; in DoCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/ |
| D | batchnorm.cc | 59 const auto &input_offset = inputs[2]; in Train() local
147 const auto &input_offset = inputs[2]; in Expand() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | random_shuffle_cpu_kernel.cc | 120 size_t input_offset = perm[i] * copy_size; in TensorShuffle() local
146 size_t input_offset = offset + perm[i] * copy_size; in TensorShuffleWithBatchRank() local
|
| D | deformable_offsets_grad_cpu_kernel.cc | 156 … const T *input_offset, const T *input_grad, T *output_grad_x, T *output_grad_offset) { in DeformableOffsetGradKernel()
264 … const T *input_x, const T *input_offset, in DeformableOffsetGradNHWCKernel()
322 … const T *input_x, const T *input_offset, in DeformableOffsetGradNCHWKernel()
405 const T *input_offset = GetDeviceAddress<T>(inputs, kOffsetIndex); in LaunchKernel() local
|
| D | shift_cpu_kernel.cc | 140 size_t input_offset = offset + LongToSize(copy_src_begin_ * inner_size); in LaunchKernel() local
|
| D | col2im_cpu_kernel.cc | 106 int64_t input_offset = batch_input_size * elt; in LaunchKernel() local
|
| D | gather_d_grad_cpu_kernel.cc | 39 size_t input_offset = 0; in GatherDGradCopyTask() local
|
| /third_party/mesa3d/src/gallium/auxiliary/draw/ |
| D | draw_prim_assembler.c | 124 unsigned input_offset = asmblr->input_verts->stride * idx; in copy_verts() local
140 unsigned input_offset = asmblr->input_verts->stride * idx; in inject_primid() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/ |
| D | custom_gather_d_grad_v2_fp32.cc | 41 int input_offset = 0; in GatherDGradCopyTask() local
|
| /third_party/skia/third_party/externals/brotli/java/org/brotli/wrapper/dec/ |
| D | decoder_jni.cc | 19 size_t input_offset; member
|
| /third_party/skia/third_party/externals/brotli/java/org/brotli/wrapper/enc/ |
| D | encoder_jni.cc | 19 size_t input_offset; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | arg_min_max_fp16.c | 52 size_t input_offset = output_offset * axis_count; in ArgMaxTopK1Fp16() local
82 size_t input_offset = output_offset * axis_count; in ArgMinTopK1Fp16() local
|
| /third_party/mesa3d/src/gallium/auxiliary/translate/ |
| D | translate.h | 73 unsigned input_offset:24; member
|