Searched refs:tensor_rhs_info (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/kernels/gemm/ |
| D | ClGemmHelpers.cpp | 58 const TensorInfo tensor_rhs_info(TensorShape(n, k, b), 1, data_type); in select_lhs_rhs_info() local
59 …const TensorShape shape = misc::shape_calculator::compute_rhs_reshaped_shape(tensor_rhs_info, info… in select_lhs_rhs_info()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/gemm/reshaped/ |
| D | ClGemmDefaultConfigReshapedBifrost.cpp | 293 const TensorInfo tensor_rhs_info(TensorShape(n, k, b), 1, DataType::F32); in configure_G76_f32() local
294 const TensorShape shape = compute_rhs_reshaped_shape(tensor_rhs_info, rhs_info_img); in configure_G76_f32()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/gemm/reshaped_only_rhs/ |
| D | ClGemmDefaultConfigReshapedRhsOnlyBifrost.cpp | 201 const TensorInfo tensor_rhs_info(TensorShape(n, k, b), 1, DataType::F32); in configure_G76_f32() local
202 const TensorShape shape = compute_rhs_reshaped_shape(tensor_rhs_info, rhs_info_img); in configure_G76_f32()
|