Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h85 int thread_end, const CpuBackendContext& cpu_backend_context) { in FullyConnectedSparseWeight1x16Impl() argument
92 const int batches = thread_end - thread_start; in FullyConnectedSparseWeight1x16Impl()
121 int thread_end, const CpuBackendContext& cpu_backend_context) { in FullyConnectedSparseWeight1x4Impl() argument
130 const int batches = thread_end - thread_start; in FullyConnectedSparseWeight1x4Impl()
144 for (int b = thread_start; b < thread_end; ++b) { in FullyConnectedSparseWeight1x4Impl()
161 int thread_end, const CpuBackendContext& cpu_backend_context_x) in FullyConnectedSparseWeight1x4Task()
173 thread_end(thread_end), in FullyConnectedSparseWeight1x4Task()
180 thread_end, cpu_backend_context); in Run()
195 int thread_end; member
250 int thread_end = thread_start + batches / thread_count; in FullyConnectedSparseWeight1x4() local
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h49 const int32_t* input_offsets, int thread_start, int thread_end, in DepthwiseConvHybridGeneral() argument
155 TFLITE_DCHECK_LE(thread_end, batches); in DepthwiseConvHybridGeneral()
157 batch_end = thread_end; in DepthwiseConvHybridGeneral()
162 TFLITE_DCHECK_LE(thread_end, output_rows); in DepthwiseConvHybridGeneral()
164 row_end = thread_end; in DepthwiseConvHybridGeneral()
268 int thread_start, int thread_end, int thread_dim) { in DepthwiseConvHybridWithRounding() argument
306 thread_start, thread_end, thread_dim); in DepthwiseConvHybridWithRounding()
317 thread_start, thread_end, thread_dim); in DepthwiseConvHybridWithRounding()
327 int thread_start, int thread_end, int thread_dim) { in DepthwiseConvHybridImpl() argument
333 thread_start, thread_end, thread_dim); in DepthwiseConvHybridImpl()
[all …]
Ddepthwise_conv.h1617 int thread_start, int thread_end, int thread_dim) {
1734 TFLITE_DCHECK_LE(thread_end, batches);
1736 batch_end = thread_end;
1741 TFLITE_DCHECK_LE(thread_end, output_rows);
1743 row_end = thread_end;
1810 int thread_start, int thread_end, int thread_dim,
1851 thread_end, thread_dim);
1876 output_data, thread_start, thread_end, thread_dim);
1885 output_data, thread_start, thread_end, thread_dim);
1894 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.h43 int thread_start, int thread_end, int thread_dim) in DepthwiseConvWorkerTask()
55 thread_end_(thread_end), in DepthwiseConvWorkerTask()
174 int thread_end = in DepthwiseConv() local
178 output_data, cpu_flags, thread_start, thread_end, in DepthwiseConv()
180 thread_start = thread_end; in DepthwiseConv()
Ddepthwiseconv_uint8.h1707 uint8* output_data, int thread_start, int thread_end, int thread_dim) {
1834 TFLITE_DCHECK_LE(thread_end, batches);
1836 batch_end = thread_end;
1842 TFLITE_DCHECK_LE(thread_end, output_height);
1844 row_end = thread_end;
2042 int thread_end, int thread_dim) {
2077 thread_end, thread_dim);
2100 bias_data, output_shape, output_data, thread_start, thread_end,
2110 thread_start, thread_end, thread_dim);
2119 int thread_end, int thread_dim) {
[all …]
Ddepthwiseconv_float.h908 int thread_end, int thread_dim) {
1015 TFLITE_DCHECK_LE(thread_end, batches);
1017 batch_end = thread_end;
1023 TFLITE_DCHECK_LE(thread_end, output_height);
1025 row_end = thread_end;
Ddepthwiseconv_uint8_3x3_filter.h5592 uint8* output_data, int thread_start, int thread_end, int thread_dim) {
5686 TFLITE_DCHECK_LE(thread_end, batches);
5688 batch_end = thread_end;
5692 TFLITE_DCHECK_LE(thread_end, params.output_height);
5694 row_end = thread_end;
12872 int thread_start, int thread_end, int thread_dim) {
12959 TFLITE_DCHECK_LE(thread_end, batches);
12961 batch_end = thread_end;
12965 TFLITE_DCHECK_LE(thread_end, output_height);
12967 row_end = thread_end;
[all …]
Dlegacy_optimized_ops.h219 uint8* output_data, int thread_start, int thread_end, int thread_dim) { in LegacyDepthwiseConvWithRounding() argument
255 bias_data, output_shape, output_data, thread_start, thread_end, in LegacyDepthwiseConvWithRounding()
265 thread_start, thread_end, thread_dim); in LegacyDepthwiseConvWithRounding()
273 uint8* output_data, int thread_start, int thread_end, int thread_dim) { in LegacyDepthwiseConvImpl() argument
277 bias_data, output_shape, output_data, thread_start, thread_end, in LegacyDepthwiseConvImpl()
388 int thread_end, int thread_dim) in LegacyDepthwiseConvWorkerTask()
399 thread_end_(thread_end), in LegacyDepthwiseConvWorkerTask()
478 int thread_end = variable
483 thread_end, thread_dim);
484 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/cronet/base/task/sequence_manager/test/
Dfake_task.cc38 ThreadTicks thread_end) in FakeTaskTiming() argument
42 end_thread_time_ = thread_end; in FakeTaskTiming()
Dfake_task.h27 ThreadTicks thread_end);
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected.cc583 int thread_end, TfLiteTensor* input_quantized, in EvalSparseHybridImpl() argument
595 const int batch_size = thread_end - thread_start; in EvalSparseHybridImpl()
672 const TfLiteTensor* bias, const int thread_start, const int thread_end, in SparseHybridFullyConnectedTask()
684 thread_end(thread_end), in SparseHybridFullyConnectedTask()
694 thread_start, thread_end, input_quantized, in Run()
708 const int thread_end; member
776 int thread_end = thread_start + batches / thread_count; in EvalHybrid() local
777 if (i < batches % thread_count) thread_end++; in EvalHybrid()
780 thread_start, thread_end, input_quantized, in EvalHybrid()
783 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()