Searched defs:thread_dim (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| H A D | depthwise_conv_hybrid.h | 50 int thread_dim) { in DepthwiseConvHybridGeneral()
268 int thread_start, int thread_end, int thread_dim) { in DepthwiseConvHybridWithRounding()
327 int thread_start, int thread_end, int thread_dim) { in DepthwiseConvHybridImpl()
391 int thread_dim; member
411 int thread_dim, thread_count, thread_dim_size; in DepthwiseConvHybridPerChannel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | device_description.cc | 67 const ThreadDim &thread_dim = thread_dim_limit(); in ToMap() local
118 const ThreadDim &thread_dim) { in ThreadDimOk()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | depthwiseconv_multithread.h | 159 int thread_dim, thread_dim_size; in DepthwiseConv() local
|
| H A D | legacy_optimized_ops.h | 219 uint8* output_data, int thread_start, int thread_end, int thread_dim) { in LegacyDepthwiseConvWithRounding()
273 uint8* output_data, int thread_start, int thread_end, int thread_dim) { in LegacyDepthwiseConvImpl()
426 int thread_dim) { in HowManyConvThreads()
454 int thread_dim, thread_count, thread_dim_size; variable
556 int thread_dim, thread_count, thread_dim_size; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | depthwiseconv_quantized_test.cc | 132 int thread_start, int thread_end, int thread_dim) { in DispatchDepthwiseConvGeneral()
147 int thread_end, int thread_dim) { in DispatchDepthwiseConvGeneral()
|