Searched refs:_gemm_output_3d (Results 1 – 2 of 2) sorted by relevance
95 …nel(), _reshape_kernel(), _im2col_output(), _weights_reshaped(), _gemm_output(), _gemm_output_3d(), in CpuGemmConv2d()345 _gemm_output_3d = TensorInfo(_gemm_output); in configure()352 _gemm_output_3d = TensorInfo(*dst); in configure()353 …_gemm_output_3d.set_data_type(output_data_type).set_data_layout(src->data_layout()).set_is_resizab… in configure()354 _gemm_output = TensorInfo(_gemm_output_3d); in configure()357 gemm_output_to_use = &_gemm_output_3d; in configure()578 _gemm_output_3d.extend_padding(out_to_use->info()->padding()); in run()579 gemm3d.allocator()->soft_init(_gemm_output_3d); in run()
229 TensorInfo _gemm_output_3d; variable