Home
last modified time | relevance | path

Searched defs:input_offset (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dcumsum_fp32.c55 int input_offset = exclusive ? 0 : 1; in Cumsum() local
115 int input_offset = exclusive ? 0 : 1; in CumsumReverse() local
176 int input_offset = exclusive ? 0 : 1; in CumsumInt() local
221 int input_offset = exclusive ? 0 : 1; in CumsumReverseInt() local
Darg_min_max_fp32.c53 int input_offset = output_offset * axis_count; in ArgMaxTopK1() local
83 int input_offset = output_offset * axis_count; in ArgMinTopK1() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dreduce_grad.c34 size_t input_offset = GetInputOffset(input_num_dims, input_dims, input_iter); in ReduceMeanByAxes() local
84 size_t input_offset = GetInputOffset(num_dims, input_dims, input_iter); in ReduceSumByAxes() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dargmax_impl.cu28 size_t input_offset = x * bound * inner_size + 0 * inner_size + y; in Argmax() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dshift_cpu_kernel.cc121 size_t input_offset = offset + copy_src_begin_ * inner_size; in Launch() local
Dgather_d_grad_cpu_kernel.cc41 size_t input_offset = 0; in GatherDGradCopyTask() local
Dgather_d_cpu_kernel.cc40 size_t input_offset = 0; in CopyTask() local
/third_party/mesa3d/src/gallium/auxiliary/draw/
Ddraw_prim_assembler.c122 unsigned input_offset = asmblr->input_verts->stride * idx; in copy_verts() local
138 unsigned input_offset = asmblr->input_verts->stride * idx; in inject_primid() local
/third_party/skia/third_party/externals/brotli/java/org/brotli/wrapper/dec/
Ddecoder_jni.cc19 size_t input_offset; member
/third_party/skia/third_party/externals/brotli/java/org/brotli/wrapper/enc/
Dencoder_jni.cc19 size_t input_offset; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Darg_min_max_fp16.c52 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/
Dtranslate.h73 unsigned input_offset:24; member
Dtranslate_generic.c56 unsigned input_offset; member
/third_party/boost/libs/compute/include/boost/compute/algorithm/detail/
Dradix_sort.hpp339 uint_ input_offset = static_cast<uint_>(first.get_index()); in radix_sort_impl() local
/third_party/boost/boost/compute/algorithm/detail/
Dradix_sort.hpp339 uint_ input_offset = static_cast<uint_>(first.get_index()); in radix_sort_impl() local
/third_party/ffmpeg/libavfilter/
Daf_afir.h41 int input_offset; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Darg_min_max_int8.c43 int input_offset = output_offset * axis_count; in DoArgMinMaxQuant() local
/third_party/mesa3d/src/freedreno/ir3/
Dir3.h372 unsigned input_offset; member
/third_party/gstreamer/gstplugins_base/gst-libs/gst/video/
Dgstvideodecoder.c385 guint64 input_offset; member