Lines Matching refs:bias_shape
96 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape…
111 …_target = compute_target(input_shape, weights_shape, bias_shape, output_shape, info, reshape_we…
112 …_reference = compute_reference(input_shape, weights_shape, bias_shape, output_shape, info, dilatio…
208 …TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, TensorShape out…
243 …TensorType bias = create_tensor<TensorType>(bias_shape, _bias_data_type, 1, _quantization_info,…
309 …ape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const TensorSha…
324 SimpleTensor<TBias> bias{ bias_shape, bias_dt, 1, _quantization_info };
366 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup() argument
369 …, AccessorType, FunctionType, T, T>::setup(input_shape, weights_shape, bias_shape, output_shape, i… in setup()
380 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape…
383 …, AccessorType, FunctionType, T, T>::setup(input_shape, weights_shape, bias_shape, output_shape, i…
394 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup() argument
397 …, AccessorType, FunctionType, T, T>::setup(input_shape, weights_shape, bias_shape, output_shape, i… in setup()
407 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup() argument
417 … AccessorType, FunctionType, T, TW>::setup(input_shape, weights_shape, bias_shape, output_shape, i… in setup()
518 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup() argument
528 compute_target(input_shape, weights_shape, bias_shape, output_shape, info, dilation); in setup()
529 compute_reference(input_shape, weights_shape, bias_shape, output_shape, info, dilation); in setup()
564 …TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, TensorShape out… in compute_target() argument
575 const auto bias_tensor_info = TensorInfo(bias_shape, 1, _data_type, _data_layout); in compute_target()
593 …ape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const TensorSha… in compute_reference() argument
596 ARM_COMPUTE_UNUSED(input_shape, weights_shape, bias_shape, output_shape, info, in compute_reference()
602 SimpleTensor<ScalarType> bias{ bias_shape, _data_type }; in compute_reference()