Searched refs:input_step (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/ |
| D | ConvDwFp32Row_sse.c | 22 size_t output_channel, size_t input_step) { in ConvDwFp32Row() argument
83 input_ptr += input_step; in ConvDwFp32Row()
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_cell_wrapper.py | 67 input_step = Tensor(np.array([1000]), mstype.int32)
68 _cell_graph_executor.compile(update_global_step, input_step)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | transpose_impl_opt.cu | 124 int input_step = NumRowsPerLoadLoop * input_dims[2]; in Swap3DTensorLast2DimKernel_shared() local
132 input_idx += input_step; // calculate the next input idx this thread should load in Swap3DTensorLast2DimKernel_shared()
139 input_idx += input_step; in Swap3DTensorLast2DimKernel_shared()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.h | 27 size_t input_channel, size_t input_step);
|
| D | conv_depthwise_fp16.c | 23 size_t output_channel, size_t input_step) { in ConvDwFp16Row() argument
28 input_ptr += input_step; in ConvDwFp16Row()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | common_func_fp32.h | 66 size_t output_channel, size_t input_step);
|
| D | conv_depthwise_fp32.c | 25 int output_channel, int input_step) { in ConvDwFp32Row() argument
30 input_ptr += input_step; in ConvDwFp32Row()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | common_func_int8.h | 35 int output_channel, int input_step, int8_t input_zp);
|
| D | conv_depthwise_int8.c | 25 int output_channel, int input_step, int8_t input_zp) { in ConvDwInt8Row() argument
31 input_ptr += input_step; in ConvDwInt8Row()
|
| /third_party/mbedtls/include/psa/ |
| D | crypto_extra.h | 1726 #define PSA_PAKE_INPUT_SIZE(alg, primitive, input_step) 0 argument
|