Lines Matching refs:shape_output
95 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs…
104 TensorType output = create_tensor<TensorType>(shape_output, data_type_output, 1);
126 …alse, false, reshape_b_only_on_first_run, (reinterpret_output_as_3d ? shape_output[2] : 0), reinte…
174 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs…
221 …transpose_A ? a_transposed : a), (pretranspose_B ? b_transposed : b), shape_output, a_offset, b_of…
226 …transpose_A ? a_transposed : a), (pretranspose_B ? b_transposed : b), shape_output, a_offset, b_of…
235 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup() argument
237 _target = compute_target(shape_a, shape_b, shape_output, a_offset, b_offset); in setup()
238 _reference = compute_reference(shape_a, shape_b, shape_output, a_offset, b_offset); in setup()
242 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs… in compute_target() argument
244 …_3d, reinterpret_output_as_3d, int32_t, false, run_twice>(shape_a, shape_b, shape_output, a_offset, in compute_target()
248 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs… in compute_reference() argument
250 …t_as_3d, uint8_t, uint8_t, false, false, run_twice>(shape_a, shape_b, shape_output, a_offset, b_of… in compute_reference()
262 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup() argument
282 …_reference = compute_reference(shape_a, shape_b, shape_output, a_offset, 0, output_stage, data_typ… in setup()
283 …_target = compute_target(shape_a, shape_b, shape_output, a_offset, 0, output_stage, data_type_a… in setup()
287 …_reference = compute_reference(shape_a, shape_b, shape_output, a_offset, b_offset, output_stage, d… in setup()
288 …_target = compute_target(shape_a, shape_b, shape_output, a_offset, b_offset, output_stage, data… in setup()
293 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs…
296 …3d, reinterpret_output_as_3d, qasymm8_t, true, run_twice>(shape_a, shape_b, shape_output, a_offset,
301 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs… in compute_reference() argument
304 …rpret_input_as_3d, TI, TW, false, false, run_twice>(shape_a, shape_b, shape_output, a_offset, b_of… in compute_reference()
336 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup() argument
339 shape_output, a_offset, b_offset, output_stage, data_type_b, false); in setup()