Home
last modified time | relevance | path

Searched defs:cpu_backend_context (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_threadpool.h38 CpuBackendContext* cpu_backend_context) { in Execute()
Dcpu_backend_context.cc112 auto* cpu_backend_context = static_cast<CpuBackendContext*>( in GetFromContext() local
Dunidirectional_sequence_gru.cc38 tflite::CpuBackendContext* cpu_backend_context) { in GruImpl()
261 auto cpu_backend_context = CpuBackendContext::GetFromContext(context); in Eval() local
Dcpu_backend_gemm_test.cc260 CpuBackendContext* cpu_backend_context) { in PerformGemmThenCompareResultsThenAgainWithClamping()
317 CpuBackendContext* cpu_backend_context) { in BisectReasonableMultiplierExponent()
370 CpuBackendContext cpu_backend_context; in TestSomeGemm() local
511 CpuBackendContext cpu_backend_context; in TestMaybeValidGemm() local
Dgru_cell.cc44 tflite::CpuBackendContext* cpu_backend_context) { in GruCell()
Dmirror_pad.cc201 CpuBackendContext* cpu_backend_context = in Eval() local
Dfully_connected.cc639 CpuBackendContext* cpu_backend_context = in EvalHybrid() local
691 CpuBackendContext* cpu_backend_context) { in FullyConnectedInt8()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h82 int thread_end, const CpuBackendContext& cpu_backend_context) { in FullyConnectedSparseWeight1x4Impl()
157 const CpuBackendContext& cpu_backend_context; member
170 CpuBackendContext* cpu_backend_context) { in FullyConnectedSparseWeight1x4()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h35 CpuBackendContext* cpu_backend_context) { in FullyConnected()
Dconv.h38 CpuBackendContext* cpu_backend_context) { in ConvPerChannel()
Dtranspose_conv.h31 int32_t* scratch_data, CpuBackendContext* cpu_backend_context) { in TransposeConvV2()
Dmean.h183 float output_scale, CpuBackendContext* cpu_backend_context) { in Mean()
Ddepthwise_conv_hybrid.h399 CpuBackendContext* cpu_backend_context) { in DepthwiseConvHybridPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h125 CpuBackendContext* cpu_backend_context) { in DepthwiseConv()
Doptimized_ops.h335 float* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
371 uint8* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
434 int16* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
815 CpuBackendContext* cpu_backend_context) { in ShuffledFullyConnected()
1131 float output_scale, CpuBackendContext* cpu_backend_context) { in Mean()
1241 float* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv()
1432 bool* compute_row_sums, CpuBackendContext* cpu_backend_context) { in HybridConvPerChannel()
1539 uint8* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv()
2787 CpuBackendContext* cpu_backend_context) { in LstmCell()
2913 int16* activ_temp_data_int16, CpuBackendContext* cpu_backend_context) { in LstmCell()
[all …]
Dlegacy_optimized_ops.h2952 CpuBackendContext* cpu_backend_context) { in TransposeConvV2()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h106 uint8_t* im2col_data, void* cpu_backend_context) { in Conv()
/external/tensorflow/tensorflow/lite/tools/benchmark/
Dbenchmark_tflite_model.cc648 std::unique_ptr<tflite::CpuBackendContext> cpu_backend_context( in InitInterpreter() local
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc1165 TestCpuBackendContext* cpu_backend_context = new TestCpuBackendContext(); in TEST_F() local