| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | portable_tensor_utils.h | 140 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
144 input, bias, input_to_gate_weights, multiplier, shift, n_batch, n_input, in MatrixBatchVectorMultiplyAccumulate()
150 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
154 input, bias, input_to_gate_weights, multiplier, shift, n_batch, n_input, in MatrixBatchVectorMultiplyAccumulate()
165 const int8_t* input_to_gate_weights, in MatrixBatchVectorMultiply() argument
171 input, input_zeropoint, input_to_gate_weights, in MatrixBatchVectorMultiply()
|
| H A D | portable_tensor_utils.cc | 330 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in PortableMatrixBatchVectorMultiplyAccumulateImpl() argument
340 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in PortableMatrixBatchVectorMultiplyAccumulateImpl()
359 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in PortableMatrixBatchVectorMultiplyAccumulate() argument
363 input, bias, input_to_gate_weights, multiplier, shift, n_batch, n_input, in PortableMatrixBatchVectorMultiplyAccumulate()
369 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in PortableMatrixBatchVectorMultiplyAccumulate() argument
373 input, bias, input_to_gate_weights, multiplier, shift, n_batch, n_input, in PortableMatrixBatchVectorMultiplyAccumulate()
379 const int8_t* input_to_gate_weights, in PortableMatrixBatchVectorMultiply() argument
392 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in PortableMatrixBatchVectorMultiply()
|
| H A D | portable_tensor_utils_impl.h | 119 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift,
125 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift,
131 const int8_t* input_to_gate_weights,
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | neon_tensor_utils.h | 107 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
111 input_to_gate_weights, multiplier, shift, n_batch, n_input, in MatrixBatchVectorMultiplyAccumulate()
117 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
121 input_to_gate_weights, multiplier, shift, n_batch, n_input, in MatrixBatchVectorMultiplyAccumulate()
126 const int8_t* input_to_gate_weights, in MatrixBatchVectorMultiply() argument
132 input, input_zeropoint, input_to_gate_weights, in MatrixBatchVectorMultiply()
|
| H A D | sse_tensor_utils.h | 121 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
125 input, input_zeropoint_times_weights, input_to_gate_weights, multiplier, in MatrixBatchVectorMultiplyAccumulate()
131 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
135 input, input_zeropoint_times_weights, input_to_gate_weights, multiplier, in MatrixBatchVectorMultiplyAccumulate()
140 const int8_t* input_to_gate_weights, in MatrixBatchVectorMultiply() argument
146 input, input_zeropoint, input_to_gate_weights, in MatrixBatchVectorMultiply()
|
| H A D | neon_tensor_utils_impl.h | 92 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift,
98 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift,
|
| H A D | neon_tensor_utils.cc | 768 const int8_t* input_to_gate_weights, in NeonMatrixBatchVectorMultiplyImpl() argument
811 (int8_t*)input_to_gate_weights + row * n_input; // NOLINT in NeonMatrixBatchVectorMultiplyImpl()
1006 const int8_t* input_to_gate_weights, int32_t n_batch, in NeonCpuBackendGemm() argument
1033 cpu_backend_gemm::Gemm(lhs_params, input_to_gate_weights, rhs_params, input, in NeonCpuBackendGemm()
1039 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in NeonMatrixBatchVectorMultiplyAccumulate() argument
1043 NeonCpuBackendGemm(input, bias, input_to_gate_weights, n_batch, n_input, in NeonMatrixBatchVectorMultiplyAccumulate()
1046 NeonMatrixBatchVectorMultiplyImpl(input, bias, input_to_gate_weights, n_batch, in NeonMatrixBatchVectorMultiplyAccumulate()
1055 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in NeonMatrixBatchVectorMultiplyAccumulate() argument
1059 NeonCpuBackendGemm(input, bias, input_to_gate_weights, n_batch, n_input, in NeonMatrixBatchVectorMultiplyAccumulate()
1062 NeonMatrixBatchVectorMultiplyImpl(input, bias, input_to_gate_weights, n_batch, in NeonMatrixBatchVectorMultiplyAccumulate()
|
| H A D | sse_tensor_utils.cc | 379 const int8_t* input_to_gate_weights, int32_t n_batch, in SseCpuBackendGemm() argument
406 cpu_backend_gemm::Gemm(lhs_params, input_to_gate_weights, rhs_params, input, in SseCpuBackendGemm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | tensor_utils_test.cc | 375 const std::vector<int8_t> input_to_gate_weights = { in TEST() local
403 input_to_gate_weights.data(), multiplier, shift, in TEST()
436 const std::vector<int8_t> input_to_gate_weights = { in TEST() local
462 input_to_gate_weights.data(), /*m_rows=*/8, /*m_cols=*/32, input.data(), in TEST()
478 input_to_gate_weights.data(), /*m_rows=*/8, /*m_cols=*/32, input.data(), in TEST()
504 input_to_gate_weights.data(), /*m_rows=*/8, /*m_cols=*/32, in TEST()
529 const std::vector<int8_t> input_to_gate_weights = { in TEST() local
558 input_to_gate_weights.data(), multiplier, shift, in TEST()
578 const std::vector<int8_t> input_to_gate_weights = { in TEST() local
606 input.data(), input_zp, input_to_gate_weights.data(), multiplier, shift, in TEST()
[all …]
|
| H A D | tensor_utils.h | 106 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift,
137 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift,
|
| H A D | portable_tensor_utils.h | 264 const int8_t* input_to_gate_weights,
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.h | 35 const int8_t* input_to_gate_weights, int32_t multiplier, in MatrixBatchVectorMultiplyAccumulate() argument
45 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in MatrixBatchVectorMultiplyAccumulate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 182 const float* input, const float* input_to_gate_weights, in CalculateLstmGateFloat() argument
206 MatrixBatchVectorMultiplyAccumulate(input_to_gate_weights, input, in CalculateLstmGateFloat()
343 const int8_t* input_to_gate_weights, in CalculateLstmGateHybrid() argument
396 input_to_gate_weights, input_to_gate_weights_ledger, n_cell, n_input, in CalculateLstmGateHybrid()
401 input_to_gate_weights, n_cell, n_input, input, in CalculateLstmGateHybrid()
531 const int8_t* input, const int8_t* input_to_gate_weights, in CalculateLstmGateInteger8x8_16() argument
564 input, input_to_gate_bias, input_to_gate_weights, input_to_gate_scale_a, in CalculateLstmGateInteger8x8_16()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 41 const float* input, const float* input_to_gate_weights, in CalculateLstmGateFloat() argument
66 input_to_gate_weights, n_cell, n_input, input, n_batch, gate); in CalculateLstmGateFloat()
|