Home
last modified time | relevance | path

Searched defs:output_offset (Results 1 – 25 of 38) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dreduce_grad.c36 size_t output_offset = GetOutputOffset(input_num_dims, input_dims, input_iter, num_axes, axes); in ReduceMeanByAxes() local
86 size_t output_offset = GetOutputOffset(num_dims, input_dims, input_iter, num_axes, axes); in ReduceSumByAxes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/
Dbatch_to_space_base.c27 int output_offset = 0; in BatchToSpaceNoCropForNHWC() local
68 int output_offset = 0; in BatchToSpaceForNHWC() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dbatch_to_space_int8.c27 int64_t output_offset = 0; in BatchToSpaceNoCropForNHWCInt8() local
74 int64_t output_offset = 0; in BatchToSpaceForNHWCInt8() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dspace_to_batch_fp32.c45 int64_t output_offset = in DoSpaceToBatch() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dcol2im.cc180 void Col2imCpuKernel::InnerCompute(int64_t c_col, int64_t input_offset, int64_t output_offset, T *i… in InnerCompute()
241 int64_t output_offset = batch_output_size * elt; in Col2imCompute() local
Drandom_shuffle.cc93 auto output_offset = shuffled_idx * block_size; in RandomShuffleCompute() local
Drandom_choice_with_mask.cc64 auto *output_offset = output_coordinate; in RandomChoiceWithMaskCompute() local
Dupsample_nearest3d_grad.cc155 …int64_t output_offset = c * output_slice_size + od * output_height * output_width + oh * output_wi… in InnerCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/sequence/
Dlist_append_and_insert_grad_cpu_kernel.cc81 size_t output_offset = element_index_size * static_cast<size_t>(index); in LaunchKernel() local
Dlist_insert_cpu_kernel.cc75 size_t output_offset = element_index_size * static_cast<size_t>(index); in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Drandom_shuffle_cpu_kernel.cc119 size_t output_offset = i * copy_size; in TensorShuffle() local
145 size_t output_offset = offset + i * copy_size; in TensorShuffleWithBatchRank() local
Dshift_cpu_kernel.cc141 size_t output_offset = offset + LongToSize(copy_dst_begin_ * inner_size); in LaunchKernel() local
Dcol2im_cpu_kernel.cc107 int64_t output_offset = batch_output_size * elt; in LaunchKernel() local
Dtensor_scatter_elements_cpu_kernel.cc120 int output_offset = 0; in Scatter() local
Dadaptive_max_pool2d_cpu_kernel.cc132 size_t output_offset = i * output_hw_; in LaunchKernel() local
Dupsample_nearest_3d_grad_cpu_kernel.cc154 …int64_t output_offset = c * output_slice_size + od * output_height * output_width + oh * output_wi… in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Drandom_shuffle_impl.cu47 int64_t output_offset = perm[row] * inner_size + col; in TensorShuffleKernel() local
Dtensor_scatter_elements.cu37 int output_offset = 0; in TensorScatterElementsKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/cuda_impl/
Dfse_decode.cu63 const uint64_t output_offset = out_size * idx / ptable_size; in FSE_Decode_kernel() local
/third_party/rust/rust/compiler/rustc_span/src/
Danalyze_source_file.rs132 let output_offset = output_offset + localVariable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Darg_min_max_fp16.c51 size_t output_offset = i * after_axis_count; in ArgMaxTopK1Fp16() local
81 size_t output_offset = i * after_axis_count; in ArgMinTopK1Fp16() local
/third_party/mesa3d/src/amd/compiler/tests/
Dmain.cpp47 static size_t output_offset = 0; variable
/third_party/mesa3d/src/gallium/auxiliary/translate/
Dtranslate.h75 unsigned output_offset; member
/third_party/mesa3d/src/gallium/auxiliary/draw/
Ddraw_mesh_prim.c84 unsigned output_offset = in copy_verts() local
Ddraw_prim_assembler.c122 unsigned output_offset = in copy_verts() local

12