Searched refs:compute_reductionB_shape (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuGemmLowpMatrixMultiplyCore.cpp | 235 _vector_sum_row = TensorInfo(compute_reductionB_shape(*a_to_use), 1, DataType::S32); in configure()
453 info_vector_sum_row = TensorInfo(compute_reductionB_shape(*a), 1, DataType::S32); in validate()
|
| /external/ComputeLibrary/examples/gemm_tuner/ |
| D | cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp | 290 … const TensorInfo info_vector_sum_row(compute_reductionB_shape(*lhs.info()), 1, DataType::S32); in do_setup()
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemmLowpMatrixMultiplyCore.cpp | 346 _vector_sum_row = TensorInfo(compute_reductionB_shape(*a), 1, DataType::S32); in configure()
547 info_vector_sum_row = TensorInfo(compute_reductionB_shape(*a), 1, DataType::S32); in validate()
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 351 inline TensorShape compute_reductionB_shape(const ITensorInfo &a) in compute_reductionB_shape() function
|