Home
last modified time | relevance | path

Searched refs:cpu_backend_context (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_threadpool.h38 CpuBackendContext* cpu_backend_context) { in Execute() argument
39 TFLITE_DCHECK_LE(tasks_count, cpu_backend_context->max_num_threads()); in Execute()
40 cpu_backend_context->ruy_context()->mutable_thread_pool()->Execute( in Execute()
50 CpuBackendContext* cpu_backend_context) {
51 TFLITE_DCHECK_LE(tasks_count, cpu_backend_context->max_num_threads());
52 cpu_backend_context->gemmlowp_context()->workers_pool()->Execute(tasks_count,
Dcpu_backend_context.cc111 auto* cpu_backend_context = static_cast<CpuBackendContext*>( in GetFromContext() local
113 if (cpu_backend_context == nullptr) { in GetFromContext()
116 cpu_backend_context = new CpuBackendContext(); in GetFromContext()
117 cpu_backend_context->SetMaxNumThreads(context->recommended_num_threads); in GetFromContext()
119 std::unique_ptr<TfLiteInternalBackendContext>(cpu_backend_context)); in GetFromContext()
122 return cpu_backend_context; in GetFromContext()
Dcpu_backend_gemm_test.cc260 CpuBackendContext* cpu_backend_context) { in PerformGemmThenCompareResultsThenAgainWithClamping() argument
263 dst_data->data(), params, cpu_backend_context); in PerformGemmThenCompareResultsThenAgainWithClamping()
276 dst_data->data(), params_with_clamp, cpu_backend_context); in PerformGemmThenCompareResultsThenAgainWithClamping()
285 dst_data->data(), params_with_clamp, cpu_backend_context); in PerformGemmThenCompareResultsThenAgainWithClamping()
317 CpuBackendContext* cpu_backend_context) { in BisectReasonableMultiplierExponent() argument
332 dst_data->data(), params_copy, cpu_backend_context); in BisectReasonableMultiplierExponent()
337 dst_params, dst_data, params_copy, cpu_backend_context); in BisectReasonableMultiplierExponent()
341 dst_params, dst_data, params_copy, cpu_backend_context); in BisectReasonableMultiplierExponent()
370 CpuBackendContext cpu_backend_context; in TestSomeGemm() local
372 cpu_backend_context.SetMaxNumThreads(1 + (random_engine() % 8)); in TestSomeGemm()
[all …]
Dmirror_pad.cc201 CpuBackendContext* cpu_backend_context = in Eval() local
203 const int thread_count = cpu_backend_context->max_num_threads(); in Eval()
226 cpu_backend_context); in Eval()
DBUILD336 name = "cpu_backend_context",
338 "cpu_backend_context.cc",
341 "cpu_backend_context.h",
383 ":cpu_backend_context",
400 ":cpu_backend_context",
428 ":cpu_backend_context",
450 ":cpu_backend_context",
647 ":cpu_backend_context",
757 ":cpu_backend_context",
1756 ":cpu_backend_context",
Dfully_connected.cc588 CpuBackendContext* cpu_backend_context = in EvalHybrid() local
600 const int max_threads = cpu_backend_context->max_num_threads(); in EvalHybrid()
628 cpu_backend_context); in EvalHybrid()
637 CpuBackendContext* cpu_backend_context) { in FullyConnectedInt8() argument
660 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() argument
135 cpu_backend_context(cpu_backend_context_x) {} in FullyConnectedSparseWeight1x4Task()
141 thread_end, cpu_backend_context); in Run()
157 const CpuBackendContext& cpu_backend_context; member
170 CpuBackendContext* cpu_backend_context) { in FullyConnectedSparseWeight1x4() argument
174 const int max_threads = cpu_backend_context->max_num_threads(); in FullyConnectedSparseWeight1x4()
182 *cpu_backend_context); in FullyConnectedSparseWeight1x4()
197 *cpu_backend_context); in FullyConnectedSparseWeight1x4()
201 cpu_backend_context); in FullyConnectedSparseWeight1x4()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dgru_cell.cc44 tflite::CpuBackendContext* cpu_backend_context) { in GruCell() argument
65 activation, cpu_backend_context); in GruCell()
82 output_shape, output, cpu_backend_context); in GruCell()
Dunidirectional_sequence_gru.cc36 tflite::CpuBackendContext* cpu_backend_context) { in GruImpl() argument
72 cpu_backend_context); in GruImpl()
253 auto cpu_backend_context = CpuBackendContext::GetFromContext(context); in Eval() local
258 cpu_backend_context); in Eval()
DBUILD83 "//tensorflow/lite/kernels:cpu_backend_context",
107 "//tensorflow/lite/kernels:cpu_backend_context",
Dgru_cell.h39 tflite::CpuBackendContext* cpu_backend_context);
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h125 CpuBackendContext* cpu_backend_context) { in DepthwiseConv() argument
133 const int max_threads = cpu_backend_context->max_num_threads(); in DepthwiseConv()
181 cpu_backend_context); in DepthwiseConv()
Doptimized_ops.h337 float* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected() argument
365 cpu_backend_context); in FullyConnected()
373 uint8* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected() argument
428 cpu_backend_context); in FullyConnected()
436 int16* output_data, CpuBackendContext* cpu_backend_context) { in FullyConnected() argument
489 cpu_backend_context); in FullyConnected()
817 CpuBackendContext* cpu_backend_context) { in ShuffledFullyConnected() argument
905 LegacyHowManyThreads<kKernelRows>(cpu_backend_context->max_num_threads(), in ShuffledFullyConnected()
937 cpu_backend_context); in ShuffledFullyConnected()
1133 float output_scale, CpuBackendContext* cpu_backend_context) { in Mean() argument
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h35 CpuBackendContext* cpu_backend_context) { in FullyConnected() argument
87 cpu_backend_context); in FullyConnected()
Dtranspose_conv.h31 int32_t* scratch_data, CpuBackendContext* cpu_backend_context) { in TransposeConvV2() argument
88 col2im_data, gemm_params, cpu_backend_context); in TransposeConvV2()
Dconv.h38 CpuBackendContext* cpu_backend_context) { in ConvPerChannel() argument
123 cpu_backend_context); in ConvPerChannel()
Dmean.h183 float output_scale, CpuBackendContext* cpu_backend_context) { in Mean() argument
219 std::min(thread_count, cpu_backend_context->max_num_threads()); in Mean()
242 cpu_backend_context); in Mean()
Ddepthwise_conv_hybrid.h399 CpuBackendContext* cpu_backend_context) { in DepthwiseConvHybridPerChannel() argument
420 const int max_threads = cpu_backend_context->max_num_threads(); in DepthwiseConvHybridPerChannel()
447 cpu_backend_context); in DepthwiseConvHybridPerChannel()
Ddepthwise_conv.h1801 const CpuBackendContext& cpu_backend_context) {
1885 const CpuBackendContext& cpu_backend_context) {
1889 output_data, thread_start, thread_end, thread_dim, cpu_backend_context);
1918 cpu_backend_context(cpu_backend_context_x) {}
1924 thread_end_, thread_dim_, cpu_backend_context);
1942 const CpuBackendContext& cpu_backend_context;
1965 CpuBackendContext* cpu_backend_context) {
1986 const int max_threads = cpu_backend_context->max_num_threads();
1994 *cpu_backend_context);
2007 thread_end, thread_dim, *cpu_backend_context);
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
DBUILD277 "//tensorflow/lite/kernels:cpu_backend_context",
323 "//tensorflow/lite/kernels:cpu_backend_context",
739 "//tensorflow/lite/kernels:cpu_backend_context",
762 "//tensorflow/lite/kernels:cpu_backend_context",
882 "//tensorflow/lite/kernels:cpu_backend_context",
917 "//tensorflow/lite/kernels:cpu_backend_context",
/external/tensorflow/tensorflow/lite/tools/benchmark/
Dbenchmark_tflite_model.cc599 std::unique_ptr<tflite::CpuBackendContext> cpu_backend_context( in InitInterpreter() local
601 cpu_backend_context->SetUseCaching(true); in InitInterpreter()
602 cpu_backend_context->SetMaxNumThreads(num_threads); in InitInterpreter()
604 std::move(cpu_backend_context)); in InitInterpreter()
DBUILD160 "//tensorflow/lite/kernels:cpu_backend_context",
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h106 uint8_t* im2col_data, void* cpu_backend_context) { in Conv() argument
107 (void)cpu_backend_context; // only used in optimized code. in Conv()
/external/tensorflow/tensorflow/lite/
DAndroid.bp73 "kernels/cpu_backend_context.cc",
Dtflite_static.bp58 "kernels/cpu_backend_context.cc",

12