Home
last modified time | relevance | path

Searched refs:thread_end (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h82 int thread_end, const CpuBackendContext& cpu_backend_context) { in FullyConnectedSparseWeight1x4Impl() argument
91 const int batches = thread_end - thread_start; in FullyConnectedSparseWeight1x4Impl()
105 for (int b = thread_start; b < thread_end; ++b) { in FullyConnectedSparseWeight1x4Impl()
122 int thread_end, const CpuBackendContext& cpu_backend_context_x) in FullyConnectedSparseWeight1x4Task()
134 thread_end(thread_end), in FullyConnectedSparseWeight1x4Task()
141 thread_end, cpu_backend_context); in Run()
156 int thread_end; member
191 int thread_end = thread_start + batches / thread_count; in FullyConnectedSparseWeight1x4() local
192 if (i < batches % thread_count) thread_end++; in FullyConnectedSparseWeight1x4()
196 output_data, thread_start, thread_end, in FullyConnectedSparseWeight1x4()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h47 const int32_t* input_offsets, int thread_start, int thread_end, in DepthwiseConvHybridGeneral() argument
153 TFLITE_DCHECK_LE(thread_end, batches); in DepthwiseConvHybridGeneral()
155 batch_end = thread_end; in DepthwiseConvHybridGeneral()
160 TFLITE_DCHECK_LE(thread_end, output_rows); in DepthwiseConvHybridGeneral()
162 row_end = thread_end; in DepthwiseConvHybridGeneral()
266 int thread_start, int thread_end, int thread_dim) { in DepthwiseConvHybridWithRounding() argument
304 thread_start, thread_end, thread_dim); in DepthwiseConvHybridWithRounding()
315 thread_start, thread_end, thread_dim); in DepthwiseConvHybridWithRounding()
325 int thread_start, int thread_end, int thread_dim) { in DepthwiseConvHybridImpl() argument
331 thread_start, thread_end, thread_dim); in DepthwiseConvHybridImpl()
[all …]
Ddepthwise_conv.h1617 int thread_start, int thread_end, int thread_dim) {
1724 TFLITE_DCHECK_LE(thread_end, batches);
1726 batch_end = thread_end;
1731 TFLITE_DCHECK_LE(thread_end, output_rows);
1733 row_end = thread_end;
1800 int thread_start, int thread_end, int thread_dim,
1841 thread_end, thread_dim);
1866 output_data, thread_start, thread_end, thread_dim);
1875 output_data, thread_start, thread_end, thread_dim);
1884 int thread_start, int thread_end, int thread_dim,
[all …]
Ddepthwise_conv_hybrid_3x3_filter.h3052 int thread_start, int thread_end, int thread_dim) {
3134 TFLITE_DCHECK_LE(thread_end, batches);
3136 batch_end = thread_end;
3140 TFLITE_DCHECK_LE(thread_end, params.output_height);
3142 row_end = thread_end;
Ddepthwise_conv_3x3_filter.h2885 int thread_start, int thread_end, int thread_dim) {
2976 TFLITE_DCHECK_LE(thread_end, batches);
2978 batch_end = thread_end;
2982 TFLITE_DCHECK_LE(thread_end, params.output_height);
2984 row_end = thread_end;
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h41 int thread_start, int thread_end, int thread_dim) in DepthwiseConvWorkerTask()
53 thread_end_(thread_end), in DepthwiseConvWorkerTask()
172 int thread_end = in DepthwiseConv() local
176 output_data, cpu_flags, thread_start, thread_end, in DepthwiseConv()
178 thread_start = thread_end; in DepthwiseConv()
Ddepthwiseconv_uint8.h1664 uint8* output_data, int thread_start, int thread_end, int thread_dim) {
1791 TFLITE_DCHECK_LE(thread_end, batches);
1793 batch_end = thread_end;
1799 TFLITE_DCHECK_LE(thread_end, output_height);
1801 row_end = thread_end;
1999 int thread_end, int thread_dim) {
2034 thread_end, thread_dim);
2057 bias_data, output_shape, output_data, thread_start, thread_end,
2067 thread_start, thread_end, thread_dim);
2076 int thread_end, int thread_dim) {
[all …]
Ddepthwiseconv_float.h906 int thread_end, int thread_dim) {
1013 TFLITE_DCHECK_LE(thread_end, batches);
1015 batch_end = thread_end;
1021 TFLITE_DCHECK_LE(thread_end, output_height);
1023 row_end = thread_end;
Ddepthwiseconv_uint8_3x3_filter.h5591 uint8* output_data, int thread_start, int thread_end, int thread_dim) {
5685 TFLITE_DCHECK_LE(thread_end, batches);
5687 batch_end = thread_end;
5691 TFLITE_DCHECK_LE(thread_end, params.output_height);
5693 row_end = thread_end;
12867 int thread_start, int thread_end, int thread_dim) {
12954 TFLITE_DCHECK_LE(thread_end, batches);
12956 batch_end = thread_end;
12960 TFLITE_DCHECK_LE(thread_end, output_height);
12962 row_end = thread_end;
[all …]
Dlegacy_optimized_ops.h218 uint8* output_data, int thread_start, int thread_end, int thread_dim) { in LegacyDepthwiseConvWithRounding() argument
254 bias_data, output_shape, output_data, thread_start, thread_end, in LegacyDepthwiseConvWithRounding()
264 thread_start, thread_end, thread_dim); in LegacyDepthwiseConvWithRounding()
272 uint8* output_data, int thread_start, int thread_end, int thread_dim) { in LegacyDepthwiseConvImpl() argument
276 bias_data, output_shape, output_data, thread_start, thread_end, in LegacyDepthwiseConvImpl()
387 int thread_end, int thread_dim) in LegacyDepthwiseConvWorkerTask()
398 thread_end_(thread_end), in LegacyDepthwiseConvWorkerTask()
477 int thread_end = variable
482 thread_end, thread_dim);
483 thread_start = thread_end;
[all …]
/external/libchrome/base/task/sequence_manager/test/
Dfake_task.cc27 ThreadTicks thread_end) in FakeTaskTiming() argument
31 end_thread_time_ = thread_end; in FakeTaskTiming()
Dfake_task.h25 ThreadTicks thread_end);
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected.cc437 const TfLiteTensor* bias, int thread_start, int thread_end, in EvalHybridImpl() argument
450 const int batch_size = thread_end - thread_start; in EvalHybridImpl()
530 const int thread_start, const int thread_end, in HybridFullyConnectedTask()
543 thread_end(thread_end), in HybridFullyConnectedTask()
553 thread_start, thread_end, input_quantized, scaling_factors, in Run()
566 const int thread_end; member
618 int thread_end = thread_start + batches / thread_count; in EvalHybrid() local
619 if (i < batches % thread_count) thread_end++; in EvalHybrid()
622 thread_start, thread_end, input_quantized, in EvalHybrid()
625 thread_start = thread_end; in EvalHybrid()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc132 int thread_start, int thread_end, int thread_dim) { in DispatchDepthwiseConvGeneral() argument
135 bias_data, output_shape, output_data, thread_start, thread_end, in DispatchDepthwiseConvGeneral()
147 int thread_end, int thread_dim) { in DispatchDepthwiseConvGeneral() argument
151 output_shape, output_data, thread_start, thread_end, thread_dim); in DispatchDepthwiseConvGeneral()
/external/llvm-project/openmp/tools/archer/
Dompt-tsan.cpp971 SET_CALLBACK(thread_end); in ompt_tsan_initialize()