| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| H A D | fully_connected.h | 85 int thread_end, const CpuBackendContext& cpu_backend_context) { in FullyConnectedSparseWeight1x16Impl()
121 int thread_end, const CpuBackendContext& cpu_backend_context) { in FullyConnectedSparseWeight1x4Impl()
196 const CpuBackendContext& cpu_backend_context; member
205 CpuBackendContext* cpu_backend_context) { in FullyConnectedSparseWeight1x16()
229 CpuBackendContext* cpu_backend_context) { in FullyConnectedSparseWeight1x4()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | cpu_backend_gemm_test.cc | 260 CpuBackendContext* cpu_backend_context) { in PerformGemmThenCompareResultsThenAgainWithClamping()
317 CpuBackendContext* cpu_backend_context) { in BisectReasonableMultiplierExponent()
370 CpuBackendContext cpu_backend_context; in TestSomeGemm() local
526 CpuBackendContext cpu_backend_context; in TestMaybeValidGemm() local
|
| H A D | add_n.cc | 72 CpuBackendContext* cpu_backend_context = in Prepare() local
114 CpuBackendContext* cpu_backend_context = in EvalAddN() local
|
| H A D | unidirectional_sequence_gru.cc | 38 tflite::CpuBackendContext* cpu_backend_context) { in GruImpl()
261 auto cpu_backend_context = CpuBackendContext::GetFromContext(context); in Eval() local
|
| H A D | cpu_backend_threadpool.h | 38 CpuBackendContext* cpu_backend_context) { in Execute()
|
| H A D | fully_connected.cc | 725 CpuBackendContext* cpu_backend_context = in EvalHybrid() local
795 CpuBackendContext* cpu_backend_context) { in FullyConnectedInt8()
852 CpuBackendContext* cpu_backend_context) { in FullyConnectedPerChannelInt8()
|
| H A D | cpu_backend_context.cc | 112 auto* cpu_backend_context = static_cast<CpuBackendContext*>( in GetFromContext() local
|
| H A D | gru_cell.cc | 44 tflite::CpuBackendContext* cpu_backend_context) { in GruCell()
|
| H A D | mirror_pad.cc | 201 CpuBackendContext* cpu_backend_context = in Eval() local
|
| H A D | reduce.cc | 664 CpuBackendContext* cpu_backend_context = in ReduceAllDims() local
|
| H A D | lstm_eval.cc | 48 CpuBackendContext* cpu_backend_context) { in MatrixBatchVectorMultiplyAccumulate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| H A D | fully_connected.h | 37 DstScalar* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnectedPerChannel()
106 DstScalar* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
|
| H A D | conv.h | 39 InputScalar* im2col_data, CpuBackendContext* cpu_backend_context) { in ConvPerChannel()
|
| H A D | transpose_conv.h | 36 CpuBackendContext* cpu_backend_context) { in TransposeConvV2()
|
| H A D | mean.h | 185 float output_scale, CpuBackendContext* cpu_backend_context) { in Mean()
|
| H A D | depthwise_conv_hybrid.h | 401 CpuBackendContext* cpu_backend_context) { in DepthwiseConvHybridPerChannel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | depthwiseconv_multithread.h | 127 CpuBackendContext* cpu_backend_context) { in DepthwiseConv()
|
| H A D | optimized_ops.h | 273 float* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
309 uint8* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
372 int16* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
753 CpuBackendContext* cpu_backend_context) { in ShuffledFullyConnected()
912 float* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv()
1103 bool* compute_row_sums, CpuBackendContext* cpu_backend_context) { in HybridConvPerChannel()
1210 uint8* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv()
2462 CpuBackendContext* cpu_backend_context) { in LstmCell()
2588 int16* activ_temp_data_int16, CpuBackendContext* cpu_backend_context) { in LstmCell()
4957 float* col2im_data, CpuBackendContext* cpu_backend_context) { in TransposeConvV2()
[all …]
|
| H A D | reduce.h | 198 float output_scale, CpuBackendContext* cpu_backend_context) { in Mean()
|
| H A D | legacy_optimized_ops.h | 2954 CpuBackendContext* cpu_backend_context) { in TransposeConvV2()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | conv.h | 115 uint8_t* im2col_data, void* cpu_backend_context) { in Conv()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/benchmark/ |
| H A D | benchmark_tflite_model.cc | 678 std::unique_ptr<tflite::CpuBackendContext> cpu_backend_context( in InitInterpreter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | interpreter_test.cc | 1236 TestCpuBackendContext* cpu_backend_context = new TestCpuBackendContext(); in TEST_F() local
|