Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/gpu/cl/operators/
DClGemmConv2d.cpp252 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/
DCpuGemmConv2d.cpp336 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()