Home
last modified time | relevance | path

Searched refs:input_to_gate_weights (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h125 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
129 input, bias, input_to_gate_weights, multiplier, shift, n_batch, n_input, in MatrixBatchVectorMultiplyAccumulate()
135 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
139 input, bias, input_to_gate_weights, multiplier, shift, n_batch, n_input, in MatrixBatchVectorMultiplyAccumulate()
150 const int8_t* input_to_gate_weights, in MatrixBatchVectorMultiply() argument
156 input, input_zeropoint, input_to_gate_weights, in MatrixBatchVectorMultiply()
Dportable_tensor_utils_impl.h110 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift,
116 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift,
122 const int8_t* input_to_gate_weights,
Dportable_tensor_utils.cc295 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in PortableMatrixBatchVectorMultiplyAccumulateImpl() argument
305 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in PortableMatrixBatchVectorMultiplyAccumulateImpl()
324 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in PortableMatrixBatchVectorMultiplyAccumulate() argument
328 input, bias, input_to_gate_weights, multiplier, shift, n_batch, n_input, in PortableMatrixBatchVectorMultiplyAccumulate()
334 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in PortableMatrixBatchVectorMultiplyAccumulate() argument
338 input, bias, input_to_gate_weights, multiplier, shift, n_batch, n_input, in PortableMatrixBatchVectorMultiplyAccumulate()
344 const int8_t* input_to_gate_weights, in PortableMatrixBatchVectorMultiply() argument
357 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in PortableMatrixBatchVectorMultiply()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.h101 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
105 input, input_zeropoint_times_weights, input_to_gate_weights, multiplier, in MatrixBatchVectorMultiplyAccumulate()
111 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
115 input, input_zeropoint_times_weights, input_to_gate_weights, multiplier, in MatrixBatchVectorMultiplyAccumulate()
120 const int8_t* input_to_gate_weights, in MatrixBatchVectorMultiply() argument
126 input, input_zeropoint, input_to_gate_weights, in MatrixBatchVectorMultiply()
Dneon_tensor_utils.h92 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
96 input_to_gate_weights, multiplier, shift, n_batch, n_input, in MatrixBatchVectorMultiplyAccumulate()
102 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in MatrixBatchVectorMultiplyAccumulate() argument
106 input_to_gate_weights, multiplier, shift, n_batch, n_input, in MatrixBatchVectorMultiplyAccumulate()
111 const int8_t* input_to_gate_weights, in MatrixBatchVectorMultiply() argument
117 input, input_zeropoint, input_to_gate_weights, in MatrixBatchVectorMultiply()
Dneon_tensor_utils_impl.h92 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,
Dneon_tensor_utils.cc742 const int8_t* input_to_gate_weights, in NeonMatrixBatchVectorMultiplyImpl() argument
785 (int8_t*)input_to_gate_weights + row * n_input; // NOLINT in NeonMatrixBatchVectorMultiplyImpl()
980 const int8_t* input_to_gate_weights, int32_t n_batch, in NeonCpuBackendGemm() argument
1007 cpu_backend_gemm::Gemm(lhs_params, input_to_gate_weights, rhs_params, input, in NeonCpuBackendGemm()
1013 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in NeonMatrixBatchVectorMultiplyAccumulate() argument
1017 NeonCpuBackendGemm(input, bias, input_to_gate_weights, n_batch, n_input, in NeonMatrixBatchVectorMultiplyAccumulate()
1020 NeonMatrixBatchVectorMultiplyImpl(input, bias, input_to_gate_weights, n_batch, in NeonMatrixBatchVectorMultiplyAccumulate()
1029 const int8_t* input_to_gate_weights, int32_t multiplier, int32_t shift, in NeonMatrixBatchVectorMultiplyAccumulate() argument
1033 NeonCpuBackendGemm(input, bias, input_to_gate_weights, n_batch, n_input, in NeonMatrixBatchVectorMultiplyAccumulate()
1036 NeonMatrixBatchVectorMultiplyImpl(input, bias, input_to_gate_weights, n_batch, in NeonMatrixBatchVectorMultiplyAccumulate()
Dsse_tensor_utils.cc177 const int8_t* input_to_gate_weights, int32_t n_batch, in SseCpuBackendGemm() argument
204 cpu_backend_gemm::Gemm(lhs_params, input_to_gate_weights, rhs_params, input, in SseCpuBackendGemm()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils.h106 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,
Dtensor_utils_test.cc373 const std::vector<int8_t> input_to_gate_weights = { in TEST() local
401 input_to_gate_weights.data(), multiplier, shift, in TEST()
434 const std::vector<int8_t> input_to_gate_weights = { in TEST() local
460 input_to_gate_weights.data(), /*m_rows=*/8, /*m_cols=*/32, input.data(), in TEST()
476 input_to_gate_weights.data(), /*m_rows=*/8, /*m_cols=*/32, input.data(), in TEST()
502 input_to_gate_weights.data(), /*m_rows=*/8, /*m_cols=*/32, in TEST()
527 const std::vector<int8_t> input_to_gate_weights = { in TEST() local
556 input_to_gate_weights.data(), multiplier, shift, in TEST()
576 const std::vector<int8_t> input_to_gate_weights = { in TEST() local
604 input.data(), input_zp, input_to_gate_weights.data(), multiplier, shift, in TEST()
[all …]
Dtensor_utils_common.h160 const int8_t* input_to_gate_weights,
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc151 const float* input, const float* input_to_gate_weights, in CalculateLstmGateFloat() argument
174 input_to_gate_weights, n_cell, n_input, input, n_batch, gate); in CalculateLstmGateFloat()
301 const int8_t* input_to_gate_weights, in CalculateLstmGateHybrid() argument
354 input_to_gate_weights, input_to_gate_weights_ledger, n_cell, n_input, in CalculateLstmGateHybrid()
359 input_to_gate_weights, n_cell, n_input, input, in CalculateLstmGateHybrid()
489 const int8_t* input, const int8_t* input_to_gate_weights, in CalculateLstmGateInteger8x8_16() argument
522 input, input_to_gate_bias, input_to_gate_weights, input_to_gate_scale_a, in CalculateLstmGateInteger8x8_16()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc41 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()