Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dops_utils.cc272 size_t GetCopySize(const std::vector<int64_t> &dim_offset, const std::vector<int64_t> &start, in GetCopySize()
283 std::vector<int64_t> dim_offset; in CalDimOffset() local
294 const std::vector<int64_t> &dim_offset) { in CalOffset()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dmirror_pad_gpu_kernel.h83 int dim_offset = output_shape_.size() - kMaxIndexOffset; in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dstrided_slice_fp32.c72 int dim_offset[DIMENSION_8D - 1]; in DoStridedSliceIn8D() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dtensor_copy_slices_cpu_kernel.cc56 auto dim_offset = CalDimOffset(input_shape_); in InitOffsetAndCopySize() local
Dstridedslice_v2_cpu_kernel.cc91 int dim_offset[DIMENSION_8D - 1]; in DoStridedSliceV2Com() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dbiasadd.c98 size_t dim_offset = in_tensor->shape_size_ - add_tensor->shape_size_; in BiasAddResize() local
/third_party/mesa3d/src/imagination/rogue/
Drogue_isa.h392 unsigned dim_offset : 5; member