Skip to content

Commit 464d02b

Browse files
wl1026sunfacebook-github-bot
authored andcommitted
to make TIE quantized linear operator to fall back to nnlib signed kernel for shapes not supported by the TIE kernel.
Summary: some quantized linear shape are not supported by the TIE operator, and should fall back to use use nnlib's signed kernel. Reviewed By: hsharma35 Differential Revision: D73944439
1 parent 385d860 commit 464d02b

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)