Home
last modified time | relevance | path

Searched defs:dim_offset (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dstrided_slice_fp32.c73 int dim_offset[DIMENSION_8D - 1]; in DoStridedSliceIntFp64Bool() local
135 int dim_offset[DIMENSION_8D - 1]; in DoStridedSlice() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dtensor_copy_slices_cpu_kernel.cc48 auto dim_offset = CalDimOffset(input_shape_); in InitKernel() local
Dmirror_pad_cpu_kernel.cc131 size_t dim_offset = output_shape_.size() - 2; in LaunchKernel() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dmirror_pad_gpu_kernel.h56 int dim_offset = output_shape_.size() - 2; in Launch() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dtensor_copy_slices.cc81 auto dim_offset = CalDimOffset(input_shape_); in Init() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc962 size_t GetCopySize(const std::vector<int64_t> &dim_offset, const std::vector<int64_t> &start, in GetCopySize()
973 std::vector<int64_t> dim_offset; in CalDimOffset() local
984 const std::vector<int64_t> &dim_offset) { in CalOffset()