| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | kernel_utils.cc | 128 int32_t* accum_scratch, int32_t* row_sums, bool* compute_row_sums) { in RnnBatchStep()
155 int32_t* accum_scratch, int32_t* row_sums, bool* compute_row_sums) { in RnnBatchStep()
|
| H A D | tensor_utils.h | 66 bool* compute_row_sums, float* scaling_factor_scratch, in MatrixBatchVectorMultiplyAccumulate()
|
| H A D | tensor_utils_test.cc | 460 bool compute_row_sums = true; in TEST() local
1169 bool compute_row_sums = true; in TestPerChannelDotprodMatrixBatchVectorMultiply() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | basic_rnn.cc | 33 bool compute_row_sums = false; member
238 TfLiteTensor* row_sums, bool* compute_row_sums) { in EvalHybrid()
|
| H A D | unidirectional_sequence_rnn.cc | 33 bool compute_row_sums = false; member
275 bool* compute_row_sums) { in EvalHybrid()
|
| H A D | lstm_eval.cc | 340 void CalculateLstmGateHybrid( in CalculateLstmGateHybrid()
480 int32_t* projection_weights_row_sums, bool* compute_row_sums, in CalculateLstmOutputHybrid()
1033 inline void LstmStepHybrid( in LstmStepHybrid()
1932 TfLiteStatus EvalHybrid( in EvalHybrid()
|
| H A D | svdf.cc | 47 bool compute_row_sums = false; member
|
| H A D | lstm_eval_test.cc | 906 bool compute_row_sums = true; in TestOneHybridAsymmLSTM() local
|
| H A D | batch_matmul.cc | 67 bool compute_row_sums = false; member
|
| H A D | unidirectional_sequence_lstm.cc | 44 bool compute_row_sums = false; member
|
| H A D | fully_connected.cc | 121 bool compute_row_sums = false; member
|
| H A D | lstm.cc | 57 bool compute_row_sums; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | svdf.h | 199 int32_t* row_sums, bool* compute_row_sums) { in EvalHybridSVDF()
|
| H A D | batch_matmul.h | 114 bool* compute_row_sums) { in BatchMatMul()
|
| H A D | portable_tensor_utils.h | 80 bool* compute_row_sums, CpuBackendContext* context) { in MatrixBatchVectorMultiplyAccumulate()
|
| H A D | portable_tensor_utils.cc | 168 bool* compute_row_sums, CpuBackendContext* context) { in PortableMatrixBatchVectorMultiplyAccumulate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | batch_matmul.h | 121 bool* compute_row_sums, CpuBackendContext* context) { in BatchMatMul()
|
| H A D | sse_tensor_utils.cc | 478 bool* compute_row_sums, CpuBackendContext* context) { in SseMatrixBatchVectorMultiplyAccumulate()
|
| H A D | neon_tensor_utils.h | 60 bool* compute_row_sums, CpuBackendContext* context) { in MatrixBatchVectorMultiplyAccumulate()
|
| H A D | sse_tensor_utils.h | 63 bool* compute_row_sums, CpuBackendContext* context) { in MatrixBatchVectorMultiplyAccumulate()
|
| H A D | neon_tensor_utils.cc | 1402 bool* compute_row_sums, CpuBackendContext* context) { in NeonMatrixBatchVectorMultiplyAccumulate()
|
| H A D | optimized_ops.h | 1103 bool* compute_row_sums, CpuBackendContext* cpu_backend_context) { in HybridConvPerChannel()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| H A D | quantized.cpp | 951 void compute_row_sums(const Requantize32 &qp, unsigned int width, unsigned int height, in compute_row_sums() function
|