Home
last modified time | relevance | path

Searched refs:output_oz_offset (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dminimal_filtering_generator.c297 int output_oz_offset = out_c_block * block_stride + out_c_res; in WinogradWeightTransform() local
323 *(winograd_data + output_oz_offset + c * oc_block) = trans_out_data[in_offset + c]; in WinogradWeightTransform()
326 output_oz_offset += block_num_stride; in WinogradWeightTransform()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dwinograd_transform_fp16.c257 int output_oz_offset = out_c_block * block_stride + out_c_res; in WinogradWeightTransformFp16() local
284 *(winograd_data + output_oz_offset + c * oc_block) = trans_out_data[in_offset + c]; in WinogradWeightTransformFp16()
287 output_oz_offset += block_num_stride; in WinogradWeightTransformFp16()