File tree Expand file tree Collapse file tree
lite/backends/nnadapter/nnadapter/src/operation Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -141,6 +141,18 @@ NNADAPTER_EXPORT int ExecuteElementwise(core::Operation* operation) {
141141 input1_shape,
142142 static_cast <math::FuseCode>(fuse_code),
143143 output_data);
144+ } else if (input0_type.precision == NNADAPTER_INT64 &&
145+ input1_type.precision == NNADAPTER_INT64) {
146+ const auto input0_data = reinterpret_cast <const int64_t *>(input0_buffer);
147+ const auto input1_data = reinterpret_cast <const int64_t *>(input1_buffer);
148+ auto output_data = reinterpret_cast <int64_t *>(output_buffer);
149+ status = math::elementwise<int64_t >(eltwise_type,
150+ input0_data,
151+ input0_shape,
152+ input1_data,
153+ input1_shape,
154+ static_cast <math::FuseCode>(fuse_code),
155+ output_data);
144156 } else if (input0_type.precision == NNADAPTER_QUANT_INT8_SYMM_PER_LAYER &&
145157 input1_type.precision == NNADAPTER_QUANT_INT8_SYMM_PER_LAYER) {
146158 const auto input0_data = reinterpret_cast <const int8_t *>(input0_buffer);
You can’t perform that action at this time.
0 commit comments