| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.cc | 71 const ThreadDim &thread_dim = thread_dim_limit(); in ToMap() local
73 absl::StrCat(thread_dim.x, ",", thread_dim.y, ",", thread_dim.z); in ToMap()
130 const ThreadDim &thread_dim) { in ThreadDimOk() argument
131 const int64 total_threads = thread_dim.x * thread_dim.y * thread_dim.z; in ThreadDimOk()
141 bool ok = thread_dim.x <= limit.x && thread_dim.y <= limit.y && in ThreadDimOk()
142 thread_dim.z <= limit.z; in ThreadDimOk()
144 VLOG(2) << "thread dim " << thread_dim.ToString() in ThreadDimOk()
|
| D | device_description.h | 338 const ThreadDim &thread_dim);
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | depthwise_conv_hybrid.h | 48 int thread_dim) { in DepthwiseConvHybridGeneral() argument
77 TFLITE_DCHECK(thread_dim == 0 || thread_dim == 1); in DepthwiseConvHybridGeneral()
150 switch (thread_dim) { 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()
349 int thread_dim) in DepthwiseConvHybridWorkerTask()
364 thread_dim(thread_dim) {} in DepthwiseConvHybridWorkerTask()
[all …]
|
| D | depthwise_conv.h | 1617 int thread_start, int thread_end, int thread_dim) {
1648 TFLITE_DCHECK(thread_dim == 0 || thread_dim == 1);
1721 switch (thread_dim) {
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,
1889 output_data, thread_start, thread_end, thread_dim, cpu_backend_context);
1902 int thread_dim,
[all …]
|
| D | depthwise_conv_hybrid_3x3_filter.h | 3052 int thread_start, int thread_end, int thread_dim) {
3092 TFLITE_DCHECK(thread_dim == 0 || thread_dim == 1);
3131 switch (thread_dim) {
|
| D | depthwise_conv_3x3_filter.h | 2885 int thread_start, int thread_end, int thread_dim) {
2931 TFLITE_DCHECK(thread_dim == 0 || thread_dim == 1);
2973 switch (thread_dim) {
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | depthwiseconv_multithread.h | 41 int thread_start, int thread_end, int thread_dim) in DepthwiseConvWorkerTask()
54 thread_dim_(thread_dim) {} in DepthwiseConvWorkerTask()
157 int thread_dim, thread_dim_size; in DepthwiseConv() local
159 thread_dim = 0; in DepthwiseConv()
162 thread_dim = 1; in DepthwiseConv()
177 thread_dim); in DepthwiseConv()
|
| D | depthwiseconv_uint8.h | 1664 uint8* output_data, int thread_start, int thread_end, int thread_dim) {
1714 TFLITE_DCHECK(thread_dim == 0 || thread_dim == 1);
1787 switch (thread_dim) {
1999 int thread_end, int thread_dim) {
2034 thread_end, thread_dim);
2058 thread_dim);
2067 thread_start, thread_end, thread_dim);
2076 int thread_end, int thread_dim) {
2080 thread_dim);
|
| D | depthwiseconv_float.h | 906 int thread_end, int thread_dim) {
921 TFLITE_DCHECK(thread_dim == 0 || thread_dim == 1);
1009 switch (thread_dim) {
|
| D | depthwiseconv_uint8_3x3_filter.h | 5591 uint8* output_data, int thread_start, int thread_end, int thread_dim) {
5641 TFLITE_DCHECK(thread_dim == 0 || thread_dim == 1);
5682 switch (thread_dim) {
12867 int thread_start, int thread_end, int thread_dim) {
12905 TFLITE_DCHECK(thread_dim == 0 || thread_dim == 1);
12951 switch (thread_dim) {
13400 uint8* output_data, int thread_start, int thread_end, int thread_dim) {
13405 thread_dim);
13414 int thread_start, int thread_end, int thread_dim) {
13419 thread_dim);
|
| D | legacy_optimized_ops.h | 218 uint8* output_data, int thread_start, int thread_end, int thread_dim) { in LegacyDepthwiseConvWithRounding() argument
255 thread_dim); 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
277 thread_dim); in LegacyDepthwiseConvImpl()
387 int thread_end, int thread_dim) in LegacyDepthwiseConvWorkerTask()
399 thread_dim_(thread_dim) {} in LegacyDepthwiseConvWorkerTask()
425 int thread_dim) { in HowManyConvThreads() argument
427 const int output_units = output_shape.Dims(thread_dim); in HowManyConvThreads()
431 FlatSizeSkipDim(output_shape, thread_dim) * filter_height * filter_width; in HowManyConvThreads()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_quantized_test.cc | 132 int thread_start, int thread_end, int thread_dim) { in DispatchDepthwiseConvGeneral() argument
136 thread_dim); 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()
|