Home
last modified time | relevance | path

Searched defs:ComputeStrides (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dtile_fp32_coder.cc26 void TileFP32Coder::ComputeStrides(const int *shape, int *strides, int ndim) const { in ComputeStrides() function in mindspore::lite::micro::nnacl::TileFP32Coder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dnnacl_common.h68 static inline void ComputeStrides(const int *shape, int *strides, const int ndim) { in ComputeStrides() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Darithmetic_dynamic_fp16_coder.cc219 void ArithmeticDynamicFP16Coder::ComputeStrides(const std::vector<std::string> &shape, in ComputeStrides() function in mindspore::lite::micro::nnacl::ArithmeticDynamicFP16Coder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dreverse_sequence_cpu_kernel.cc31 void ReverseSequenceCpuKernelMod::ComputeStrides(const std::vector<int64_t> &shape, int *strides, in ComputeStrides() function in mindspore::kernel::ReverseSequenceCpuKernelMod