Searched defs:compute_row_sums (Results 1 – 22 of 22) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| 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()
|
| D | tensor_utils.h | 66 bool* compute_row_sums, float* scaling_factor_scratch, in MatrixBatchVectorMultiplyAccumulate()
|
| D | tensor_utils_test.cc | 458 bool compute_row_sums = true; in TEST() local
1167 bool compute_row_sums = true; in TestPerChannelDotprodMatrixBatchVectorMultiply() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 33 bool compute_row_sums = false; member
237 TfLiteTensor* row_sums, bool* compute_row_sums) { in EvalHybrid()
|
| D | unidirectional_sequence_rnn.cc | 33 bool compute_row_sums = false; member
275 bool* compute_row_sums) { in EvalHybrid()
|
| D | svdf.cc | 47 bool compute_row_sums = false; member
|
| D | lstm_eval.cc | 298 void CalculateLstmGateHybrid( in CalculateLstmGateHybrid()
438 int32_t* projection_weights_row_sums, bool* compute_row_sums, in CalculateLstmOutputHybrid()
977 inline void LstmStepHybrid( in LstmStepHybrid()
1868 TfLiteStatus EvalHybrid( in EvalHybrid()
|
| D | lstm_eval_test.cc | 906 bool compute_row_sums = true; in TestOneHybridAsymmLSTM() local
|
| D | batch_matmul.cc | 67 bool compute_row_sums = false; member
|
| D | unidirectional_sequence_lstm.cc | 43 bool compute_row_sums = false; member
|
| D | fully_connected.cc | 117 bool compute_row_sums = false; member
|
| D | lstm.cc | 57 bool compute_row_sums; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | svdf.h | 199 int32_t* row_sums, bool* compute_row_sums) { in EvalHybridSVDF()
|
| D | batch_matmul.h | 113 bool* compute_row_sums) { in BatchMatMul()
|
| D | portable_tensor_utils.h | 80 bool* compute_row_sums, CpuBackendContext* context) { in MatrixBatchVectorMultiplyAccumulate()
|
| D | portable_tensor_utils.cc | 168 bool* compute_row_sums, CpuBackendContext* context) { in PortableMatrixBatchVectorMultiplyAccumulate()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 121 bool* compute_row_sums, CpuBackendContext* context) { in BatchMatMul()
|
| D | sse_tensor_utils.cc | 275 bool* compute_row_sums, CpuBackendContext* context) { in SseMatrixBatchVectorMultiplyAccumulate()
|
| D | sse_tensor_utils.h | 58 bool* compute_row_sums, CpuBackendContext* context) { in MatrixBatchVectorMultiplyAccumulate()
|
| D | neon_tensor_utils.h | 60 bool* compute_row_sums, CpuBackendContext* context) { in MatrixBatchVectorMultiplyAccumulate()
|
| D | neon_tensor_utils.cc | 1376 bool* compute_row_sums, CpuBackendContext* context) { in NeonMatrixBatchVectorMultiplyAccumulate()
|
| D | optimized_ops.h | 1434 bool* compute_row_sums, CpuBackendContext* cpu_backend_context) { in HybridConvPerChannel()
|