Skip to content

Commit

Permalink
Replace uses of int64 with int64_t and similar integer type aliases.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 588698659
  • Loading branch information
gribozavr authored and tensorflower-gardener committed Dec 7, 2023
1 parent 5b5eaab commit bebd7ec
Show file tree
Hide file tree
Showing 3 changed files with 532 additions and 512 deletions.
4 changes: 2 additions & 2 deletions tensorflow/lite/kernels/cpu_backend_gemm.h
Original file line number Diff line number Diff line change
Expand Up @@ -176,7 +176,7 @@ template <QuantizationFlavor quantization_flavor>
void Gemm(const MatrixParams<int8_t>& lhs_params, const int8_t* lhs_data,
const MatrixParams<int16_t>& rhs_params, const int16_t* rhs_data,
const MatrixParams<int16_t>& dst_params, int16_t* dst_data,
const GemmParams<int32_t, int16, quantization_flavor>& params,
const GemmParams<int32_t, int16_t, quantization_flavor>& params,
CpuBackendContext* context) {
ruy::profiler::ScopeLabel label("cpu_backend_gemm::Gemm");
ValidateParams(lhs_params, rhs_params, dst_params, params);
Expand All @@ -187,7 +187,7 @@ void Gemm(const MatrixParams<int8_t>& lhs_params, const int8_t* lhs_data,

// Currently, only Ruy backend supports 16x8 quant gemm so we use ruy
// only.
detail::GemmImplUsingRuy<int8_t, int16_t, int32_t, int16,
detail::GemmImplUsingRuy<int8_t, int16_t, int32_t, int16_t,
quantization_flavor>::Run(lhs_params, lhs_data,
rhs_params, rhs_data,
dst_params, dst_data,
Expand Down
Loading

0 comments on commit bebd7ec

Please sign in to comment.