Home
last modified time | relevance | path

Searched defs:broadcast_bias (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DGEMMFixture.h234 … unsigned int k, unsigned int batch_size, float alpha, float beta, bool broadcast_bias, bool fp16_… in setup()
263 …e, const TensorShape &bias_shape, DataType data_type, float alpha, float beta, bool broadcast_bias, in compute_target()
316 …pe, const TensorShape &rhs_shape, DataType data_type, float alpha, float beta, bool broadcast_bias, in compute_reference()
358 … unsigned int k, unsigned int batch_size, float alpha, float beta, bool broadcast_bias, bool fp16_… in setup()
480 …h_size, float alpha, float beta, unsigned int v0, unsigned int h0, bool broadcast_bias, bool fp16_… in setup()
524 …DataType data_type, float alpha, float beta, bool broadcast_bias, bool fp16_mixed_precision, const… in compute_target()
594 …pe, const TensorShape &rhs_shape, DataType data_type, float alpha, float beta, bool broadcast_bias, in compute_reference()
636 …ned int batch_size, float alpha, float beta, unsigned int v0, unsigned int h0, bool broadcast_bias, in setup()
791 …l export_to_cl_image, DataType data_type, float alpha, float beta, bool broadcast_bias, bool lhs_t… in setup()
838 …DataType data_type, float alpha, float beta, bool broadcast_bias, const ActivationLayerInfo &act_i… in compute_target()
[all …]
H A DGEMMLowpFixture.h1392 … int k0, unsigned int h0, bool interleave_rhs, bool transpose_rhs, bool broadcast_bias, DataType d… in setup()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClGemm.cpp215 bool broadcast_bias = gemm_info.broadcast_bias(); in configure_native() local
247 bool broadcast_bias = gemm_info.broadcast_bias(); in configure_reshaped() local
292 bool broadcast_bias = gemm_info.broadcast_bias(); in configure_reshaped_only_rhs() local
339 bool broadcast_bias = gemm_info.broadcast_bias(); in configure_reshaped_only_rhs_mmul() local
389 const bool broadcast_bias = gemm_info.broadcast_bias(); in validate_native() local
426 const bool broadcast_bias = gemm_info.broadcast_bias(); in validate_reshaped() local
475 const bool broadcast_bias = gemm_info.broadcast_bias(); in validate_reshaped_only_rhs() local
524 const bool broadcast_bias = gemm_info.broadcast_bias(); in validate_reshaped_only_rhs_mmul() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DGEMMLowpMatrixMultiplyReshapedOnlyRhsMMUL.cpp94 const auto broadcast_bias = framework::dataset::make("broadcast_bias", {true, false}); variable
H A DGEMMMatrixMultiplyNative.cpp249 …signed int m0_value, unsigned int n0_value, unsigned int k0_value, bool broadcast_bias, DataType d… in validate_configuration()
H A DGEMMMatrixMultiplyReshapedOnlyRHS.cpp270 …bool i_value_rhs, bool t_value_rhs, bool export_to_cl_image, bool broadcast_bias, bool input_as_3d… in validate_configuration()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_rewriter.cc305 HloInstruction *broadcast_bias, in FuseVectorBiasAdd()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/
H A DTypes.h2250 bool broadcast_bias() const in broadcast_bias() function
2499 bool broadcast_bias() const in broadcast_bias() function