Skip to content

Commit ff0df1c

Browse files
authored
to make TIE quantized linear operator to fall back to nnlib signed kernel for shapes not supported by the TIE kernel.
Differential Revision: D73944439 Pull Request resolved: #10637
1 parent a4322c7 commit ff0df1c

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

backends/cadence/hifi/operators/operators.h

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,32 @@ ::executorch::aten::Tensor& div_out_mode(
3636
::executorch::aten::optional<::executorch::aten::string_view> mode,
3737
::executorch::aten::Tensor& out);
3838

39+
void quantized_linear_out(
40+
__ET_UNUSED ::executorch::runtime::KernelRuntimeContext& ctx,
41+
const ::executorch::aten::Tensor& in,
42+
const ::executorch::aten::Tensor& weight,
43+
const ::executorch::aten::Tensor& bias,
44+
int64_t in_zero_point,
45+
const ::executorch::aten::Tensor& weight_zero_point,
46+
const ::executorch::aten::Tensor& out_multiplier,
47+
const ::executorch::aten::Tensor& out_shift,
48+
int64_t out_zero_point,
49+
__ET_UNUSED const ::executorch::aten::optional<::executorch::aten::Tensor>& offset,
50+
::executorch::aten::Tensor& out);
51+
52+
void quantized_linear_per_tensor_out(
53+
__ET_UNUSED ::executorch::runtime::KernelRuntimeContext& ctx,
54+
const ::executorch::aten::Tensor& in,
55+
const ::executorch::aten::Tensor& weight,
56+
const ::executorch::aten::Tensor& bias,
57+
int64_t in_zero_point,
58+
int64_t weight_zero_point,
59+
int64_t out_multiplier,
60+
int64_t out_shift,
61+
int64_t out_zero_point,
62+
__ET_UNUSED const ::executorch::aten::optional<::executorch::aten::Tensor>& offset,
63+
::executorch::aten::Tensor& out);
64+
3965
} // namespace native
4066
} // namespace HiFi
4167
} // namespace impl

0 commit comments

Comments
 (0)