Searched refs:dim_offset (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | strided_slice_fp32.c | 73 int dim_offset[DIMENSION_8D - 1]; in DoStridedSliceIntFp64Bool() local
74 dim_offset[6] = in_shape[7]; in DoStridedSliceIntFp64Bool()
75 dim_offset[5] = in_shape[6] * dim_offset[6]; in DoStridedSliceIntFp64Bool()
76 dim_offset[4] = in_shape[5] * dim_offset[5]; in DoStridedSliceIntFp64Bool()
77 dim_offset[3] = in_shape[4] * dim_offset[4]; in DoStridedSliceIntFp64Bool()
78 dim_offset[2] = in_shape[3] * dim_offset[3]; in DoStridedSliceIntFp64Bool()
79 dim_offset[1] = in_shape[2] * dim_offset[2]; in DoStridedSliceIntFp64Bool()
80 dim_offset[0] = in_shape[1] * dim_offset[1]; in DoStridedSliceIntFp64Bool()
91 … int32_t in_offset = dim0 * dim_offset[0] + dim1 * dim_offset[1] + dim2 * dim_offset[2] + in DoStridedSliceIntFp64Bool()
92 … dim3 * dim_offset[3] + dim4 * dim_offset[4] + dim5 * dim_offset[5] + in DoStridedSliceIntFp64Bool()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | tensor_copy_slices_cpu_kernel.cc | 48 auto dim_offset = CalDimOffset(input_shape_); in InitKernel() local
50 offset_ = CalOffset(begin, end, dim_offset) * type_size; in InitKernel()
51 copy_size_ = GetCopySize(dim_offset, begin, end) * type_size; in InitKernel()
|
| D | mirror_pad_cpu_kernel.cc | 131 size_t dim_offset = output_shape_.size() - 2; in LaunchKernel() local
133 const int64_t padded_height = output_shape_[dim_offset]; in LaunchKernel()
134 const int64_t padded_width = output_shape_[dim_offset + 1]; in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | mirror_pad_gpu_kernel.h | 56 int dim_offset = output_shape_.size() - 2; in Launch() local
59 …output_shape_[dim_offset + 0], output_shape_[dim_offset + 1], num_paddings_, paddings, mode_, outp… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | tensor_copy_slices.cc | 81 auto dim_offset = CalDimOffset(input_shape_); in Init() local
82 offset_ = CalOffset(begin, end, dim_offset) * abstract::TypeIdSize(input_type_id_); in Init()
83 copy_size_ = GetCopySize(dim_offset, begin, end) * abstract::TypeIdSize(input_type_id_); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 962 size_t GetCopySize(const std::vector<int64_t> &dim_offset, const std::vector<int64_t> &start, in GetCopySize() argument
966 return SizetMulWithOverflowCheck(LongToSize(stop[i] - start[i]), LongToSize(dim_offset[i])); in GetCopySize()
969 return LongToSize(dim_offset[start.size() - 1]); in GetCopySize()
973 std::vector<int64_t> dim_offset; in CalDimOffset() local
976 dim_offset.push_back(offset); in CalDimOffset()
979 std::reverse(dim_offset.begin(), dim_offset.end()); in CalDimOffset()
980 return dim_offset; in CalDimOffset()
984 const std::vector<int64_t> &dim_offset) { in CalOffset() argument
988 offset += SizetMulWithOverflowCheck(LongToSize(dim_offset[i]), LongToSize(start[i])); in CalOffset()
|
| D | common_utils.h | 142 const std::vector<int64_t> &dim_offset);
144 size_t GetCopySize(const std::vector<int64_t> &dim_offset, const std::vector<int64_t> &start,
|