Searched defs:reduction_size (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils_impl.h | 215 int reduction_size) { in PortableReductionSumVector()
|
| D | portable_tensor_utils.h | 295 int output_size, int reduction_size) { in ReductionSumVector()
301 int output_size, int reduction_size) { in ReductionSumVector()
307 int output_size, int reduction_size) { in ReductionSumVector()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 138 int reduction_size = it->second; in Reduce() local
487 const double reduction_size = total_src_elements / total_dst_elements; in BindArguments() local
|
| D | mean_stddev_normalization.cc | 68 int reduction_size = work_group_size.z; in GetReduceCode() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | sse_tensor_utils.h | 306 int output_size, int reduction_size) { in ReductionSumVector()
312 int output_size, int reduction_size) { in ReductionSumVector()
318 int output_size, int reduction_size) { in ReductionSumVector()
|
| D | neon_tensor_utils.h | 291 int output_size, int reduction_size) { in ReductionSumVector()
297 int output_size, int reduction_size) { in ReductionSumVector()
303 int output_size, int reduction_size) { in ReductionSumVector()
|
| D | sse_tensor_utils.cc | 621 const int output_size, const int reduction_size) { in SseReductionSumVector()
|
| D | neon_tensor_utils.cc | 2566 int output_size, int reduction_size) { in NeonReductionSumVector()
2590 const int output_size, const int reduction_size) { in NeonReductionSumVector()
|