Skip to content

Commit 1daeac0

Browse files
authored
Fix windows build of grouped convolutions. (#656)
1 parent 6bb5a43 commit 1daeac0

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

larq_compute_engine/core/indirect_bgemm/select_kernel.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ namespace indirect_bgemm {
2929
template <typename DstScalar>
3030
inline std::unique_ptr<Kernel> SelectRuntimeKernel(
3131
const bconv2d::BConv2DParams* bconv2d_params,
32-
const tflite::RuntimeShape& bitpacked_input_shape,
33-
const tflite::RuntimeShape& output_shape,
32+
const ::tflite::RuntimeShape& bitpacked_input_shape,
33+
const ::tflite::RuntimeShape& output_shape,
3434
const bconv2d::OutputTransform<DstScalar>& output_transform) {
3535
static_assert(std::is_same<DstScalar, float>::value ||
3636
std::is_same<DstScalar, std::int8_t>::value,
@@ -133,8 +133,8 @@ inline std::unique_ptr<Kernel> SelectRuntimeKernel(
133133
template <>
134134
inline std::unique_ptr<Kernel> SelectRuntimeKernel<TBitpacked>(
135135
const bconv2d::BConv2DParams* bconv2d_params,
136-
const tflite::RuntimeShape& bitpacked_input_shape,
137-
const tflite::RuntimeShape& output_shape,
136+
const ::tflite::RuntimeShape& bitpacked_input_shape,
137+
const ::tflite::RuntimeShape& output_shape,
138138
const bconv2d::OutputTransform<TBitpacked>& output_transform) {
139139
// Only the C++ kernel currently supports bitpacked output.
140140
return std::make_unique<Kernel4x2Portable<TBitpacked>>(

0 commit comments

Comments
 (0)