Home
last modified time | relevance | path

Searched refs:GemmParams (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm_params.h118 struct GemmParams { struct
154 using QuantizedGemmParams = GemmParams<std::int32_t, DstScalar>; argument
156 using FloatGemmParams = GemmParams<float, float>;
170 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params) { in ValidateGemmParams()
235 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params) {
Dcpu_backend_gemm_ruy.h66 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params, in Run()
81 const GemmParams<std::int32_t, DstScalar, quantization_flavor>& params,
107 const GemmParams<std::int32_t, std::int32_t, quantization_flavor>& params,
116 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params,
129 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params,
Dcpu_backend_gemm_test.cc44 using cpu_backend_gemm::GemmParams;
140 void Clamp(const GemmParams<AccumScalar, DstScalar, quantization_flavor>& src, in Clamp()
142 GemmParams<AccumScalar, DstScalar, quantization_flavor>* dst) { in Clamp()
258 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params, in PerformGemmThenCompareResultsThenAgainWithClamping()
268 GemmParams<AccumScalar, DstScalar, quantization_flavor> params_with_clamp; in PerformGemmThenCompareResultsThenAgainWithClamping()
316 const GemmParams<AccumScalar, DstScalar>& params, in BisectReasonableMultiplierExponent()
327 GemmParams<AccumScalar, DstScalar> params_copy(params); in BisectReasonableMultiplierExponent()
351 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params, in ReferenceGemm()
432 GemmParams<AccumScalar, DstScalar> params; in TestSomeGemm()
490 GemmParams<AccumScalar, DstScalar, perchannel_flavor> params_perchannel; in TestSomeGemm()
Dcpu_backend_gemm_custom_gemv.h83 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params) { in IsSupportedGivenSufficientlyManyRows()
92 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params, in Run()
105 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params, in CustomGemvTask()
131 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params_;
151 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params, in CustomGemv()
313 const GemmParams<std::int32_t, DstScalar, quantization_flavor>& params) {
326 const GemmParams<std::int32_t, DstScalar, quantization_flavor>& params,
627 const GemmParams<float, float>& params) {
638 const GemmParams<float, float>& params, int row_start,
Dcpu_backend_gemm_x86.h42 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params, in Run()
71 const GemmParams<float, float,
Dcpu_backend_gemm_eigen.h32 const GemmParams<float, float>& params,
Dcpu_backend_gemm.h119 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params, in Gemm()
174 const GemmParams<int32_t, int32_t, quantization_flavor>& params, in Gemm()
Dcpu_backend_gemm_gemmlowp.h84 const GemmParams<std::int32_t, DstScalar,
141 const GemmParams<std::int32_t, DstScalar,
Dcpu_backend_gemm_eigen.cc41 const GemmParams<float, float>& params, CpuBackendContext* /* context */) { in Run()
/external/gemmlowp/meta/
Dlegacy_single_thread_gemm.h40 typedef GemmParams<std::uint8_t, std::uint8_t, RowMajorWithSum, in gemm_q8_strided()
84 typedef GemmParams<std::uint8_t, std::uint8_t, RowMajorWithSum, in gemv_q8()
132 typedef GemmParams<std::uint8_t, std::int32_t, RowMajorWithSum, in gemm_i32_strided()
172 typedef GemmParams<std::uint8_t, std::int32_t, RowMajorWithSum, in gemv_i32()
217 typedef GemmParams<std::uint8_t, float, RowMajorWithSum, RowMajorWithSum, in gemm_f_strided()
257 typedef GemmParams<std::uint8_t, float, RowMajorWithSum, RowMajorWithSum, in gemv_f()
Dtest_gemm_correctness.cc45 typedef GemmParams<std::uint8_t, std::uint8_t, RowMajorWithSum, ColumnMajorWithSum,
49 typedef GemmParams<std::uint8_t, std::uint8_t, RowMajorWithSum, RowMajorWithSum,
53 typedef GemmParams<std::uint8_t, float, RowMajorWithSum, ColumnMajorWithSum,
57 typedef GemmParams<std::uint8_t, float, RowMajorWithSum, RowMajorWithSum,
61 typedef GemmParams<std::uint8_t, std::int32_t, RowMajorWithSum, ColumnMajorWithSum,
65 typedef GemmParams<std::uint8_t, std::int32_t, RowMajorWithSum, RowMajorWithSum,
Dbase.h47 struct GemmParams { struct
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h33 using ::tflite::cpu_backend_gemm::GemmParams; in BatchMatMul()
107 GemmParams<float, float> gemm_params; in BatchMatMul()
123 using ::tflite::cpu_backend_gemm::GemmParams; in BatchMatMul()
226 GemmParams<int32_t, int32_t> gemm_params; in BatchMatMul()
277 using ::tflite::cpu_backend_gemm::GemmParams; in BatchMatMul()
365 GemmParams<int32_t, int8_t> gemm_params; in BatchMatMul()
Dsse_tensor_utils.cc181 using ::tflite::cpu_backend_gemm::GemmParams; in SseCpuBackendGemm()
200 GemmParams<int32, int32> gemm_params; in SseCpuBackendGemm()
Doptimized_ops.h359 cpu_backend_gemm::GemmParams<float, float> gemm_params; in FullyConnected()
420 cpu_backend_gemm::GemmParams<int32, uint8> gemm_params; in FullyConnected()
481 cpu_backend_gemm::GemmParams<int32, int16> gemm_params; in FullyConnected()
1323 cpu_backend_gemm::GemmParams<float, float> gemm_params; in Conv()
1511 cpu_backend_gemm::GemmParams<int32_t, int32_t> gemm_params; in HybridConvPerChannel()
1629 cpu_backend_gemm::GemmParams<int32, uint8> gemm_params; in Conv()
3001 cpu_backend_gemm::GemmParams<int32, int16> gemm_params; in LstmCell()
5766 cpu_backend_gemm::GemmParams<float, float> gemm_params; in TransposeConvV2()
5997 cpu_backend_gemm::GemmParams<int32_t, int32_t> gemm_params; in TransposeConvV2()
Dneon_tensor_utils.cc984 using ::tflite::cpu_backend_gemm::GemmParams; in NeonCpuBackendGemm()
1003 GemmParams<int32, int32> gemm_params; in NeonCpuBackendGemm()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h79 cpu_backend_gemm::GemmParams<int32, int8> gemm_params; in FullyConnected()
Dtranspose_conv.h85 cpu_backend_gemm::GemmParams<int32_t, int32_t> gemm_params; in TransposeConvV2()
Dconv.h112 cpu_backend_gemm::GemmParams< in ConvPerChannel()
/external/tensorflow/tensorflow/core/kernels/
Dmeta_support.cc122 typedef gemmlowp::meta::GemmParams< in QuantizedGemmImpl()