Lines Matching refs:bias_shape
240 const TensorShape bias_shape(n, in setup()
244 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, data_type, alpha, beta, broadcast_bi… in setup()
263 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, DataType data_t… in compute_target() argument
269 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
369 const TensorShape bias_shape(n, 1, 1); in setup()
371 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, data_type, alpha, beta, m_h, fp16_mi… in setup()
386 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, DataType data_t… in compute_target() argument
392 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
500 const TensorShape bias_shape(n, in setup()
504 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha… in setup()
523 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
529 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
661 const TensorShape bias_shape(n, 1, 1); in setup()
663 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha… in setup()
678 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
684 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
811 const TensorShape bias_shape(n, in setup()
815 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha, b… in setup()
837 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
843 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
999 const TensorShape bias_shape(n, in setup()
1013 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha, b… in setup()
1035 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
1041 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
1255 const TensorShape bias_shape(n, 1, 1); in setup()
1257 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha, b… in setup()
1275 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
1281 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
1429 const TensorShape bias_shape(n, in setup()
1433 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha, b… in setup()
1455 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
1461 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
1598 const TensorShape bias_shape(n, in setup()
1612 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha, b… in setup()
1634 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
1640 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
1835 const TensorShape bias_shape(n, 1, 1); in setup()
1837 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha, b… in setup()
1855 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
1862 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
1998 const TensorShape bias_shape(n, in setup()
2002 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha… in setup()
2021 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
2027 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
2137 const TensorShape bias_shape(n, in setup()
2151 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha… in setup()
2170 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
2176 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
2345 const TensorShape bias_shape(n, 1, 1); in setup()
2347 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha… in setup()
2362 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
2368 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()
2482 const TensorShape bias_shape(n, in setup()
2486 …_target = compute_target(lhs_shape, rhs_shape, bias_shape, lhs_info, rhs_info, data_type, alpha… in setup()
2505 …nsorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const GEMMLHSMa… in compute_target() argument
2511 TensorType bias = create_tensor<TensorType>(bias_shape, data_type, 1); in compute_target()