Searched refs:shape_gemm (Results 1 – 2 of 2) sorted by relevance
/external/ComputeLibrary/src/gpu/cl/operators/ |
D | ClGemmConv2d.cpp | 252 TensorShape shape_gemm; in configure() local 255 shape_gemm = _im2col_output.tensor_shape(); in configure() 256 shape_gemm.set(0, mat_weights_cols); in configure() 257 shape_gemm.set(1, conv_w * conv_h); in configure() 259 _gemm_output = TensorInfo(shape_gemm, 1, data_type); in configure() 463 TensorShape shape_gemm; in validate() local 465 shape_gemm = gemm_input_to_use->tensor_shape(); in validate() 466 shape_gemm.set(0, mat_weights_cols); in validate() 467 shape_gemm.set(1, conv_w * conv_h); in validate() 469 info_gemm = TensorInfo(shape_gemm, 1, data_type); in validate()
|
/external/ComputeLibrary/src/cpu/operators/ |
D | CpuGemmConv2d.cpp | 336 TensorShape shape_gemm; in configure() local 339 shape_gemm = _im2col_output.tensor_shape(); in configure() 340 shape_gemm.set(0, mat_weights_cols); in configure() 341 shape_gemm.set(1, conv_w * conv_h); in configure() 343 _gemm_output = TensorInfo(shape_gemm, 1, output_data_type); in configure() 524 TensorShape shape_gemm = gemm_input_to_use->tensor_shape(); in validate() local 525 shape_gemm.set(0, mat_weights_cols); in validate() 526 shape_gemm.set(1, conv_w * conv_h); in validate() 527 info_gemm = TensorInfo(shape_gemm, 1, output_data_type); in validate()
|