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.cc111 auto* cpu_backend_context = static_cast<CpuBackendContext*>( in GetFromContext() 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
Dmirror_pad.cc201 CpuBackendContext* cpu_backend_context = in Eval() local
Dfully_connected.cc588 CpuBackendContext* cpu_backend_context = in EvalHybrid() local
637 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()
Dtranspose_conv.h31 int32_t* scratch_data, CpuBackendContext* cpu_backend_context) { in TransposeConvV2()
Dconv.h38 CpuBackendContext* cpu_backend_context) { in ConvPerChannel()
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/experimental/kernels/
Dunidirectional_sequence_gru.cc36 tflite::CpuBackendContext* cpu_backend_context) { in GruImpl()
253 auto cpu_backend_context = CpuBackendContext::GetFromContext(context); in Eval() local
Dgru_cell.cc44 tflite::CpuBackendContext* cpu_backend_context) { in GruCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h125 CpuBackendContext* cpu_backend_context) { in DepthwiseConv()
Doptimized_ops.h337 float* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
373 uint8* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
436 int16* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected()
817 CpuBackendContext* cpu_backend_context) { in ShuffledFullyConnected()
1133 float output_scale, CpuBackendContext* cpu_backend_context) { in Mean()
1243 float* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv()
1434 bool* compute_row_sums, CpuBackendContext* cpu_backend_context) { in HybridConvPerChannel()
1541 uint8* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv()
2785 CpuBackendContext* cpu_backend_context) { in LstmCell()
2911 int16* activ_temp_data_int16, CpuBackendContext* cpu_backend_context) { in LstmCell()
[all …]
Dlegacy_optimized_ops.h2953 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.cc599 std::unique_ptr<tflite::CpuBackendContext> cpu_backend_context( in InitInterpreter() local
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc1160 TestCpuBackendContext* cpu_backend_context = new TestCpuBackendContext(); in TEST_F() local