Searched refs:tmp_out_data (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_transform_fp16.h | 36 void WinogradOutputNHWCTransformFp16(const float16_t *gemm_out, float16_t *tmp_out_data, const floa…
40 void WinogradOutputNC8HW8TransformFp16(const float16_t *gemm_out, float16_t *tmp_out_data, const fl…
|
| D | winograd_transform_fp16.c | 128 void WinogradOutputNHWCTransformFp16(const float16_t *gemm_out, float16_t *tmp_out_data, const floa… in WinogradOutputNHWCTransformFp16() argument
160 float16_t *dst_ptr = tmp_out_data + dst_oc8_offset; in WinogradOutputNHWCTransformFp16()
167 void WinogradOutputNC8HW8TransformFp16(const float16_t *gemm_out, float16_t *tmp_out_data, const fl… in WinogradOutputNC8HW8TransformFp16() argument
200 float16_t *dst_ptr = tmp_out_data + dst_oc8_offset; in WinogradOutputNC8HW8TransformFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 494 int8_t *tmp_out_data = out_data; in MaxPoolingOptInt8() local
496 vst1q_s8(tmp_out_data, vld1q_s8(out_array + j)); in MaxPoolingOptInt8()
497 tmp_out_data += 16; in MaxPoolingOptInt8()
501 vst1_s8(tmp_out_data, vld1_s8(out_array + j)); in MaxPoolingOptInt8()
502 tmp_out_data += 8; in MaxPoolingOptInt8()
|