Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc588 DimensionVector kernel_limits; in HandleConvolution() local
593 kernel_limits.push_back(1); in HandleConvolution()
603 kernel_limits.push_back(kernel_limit); in HandleConvolution()
629 kernel_limits[spatial_dimension] == output_limits[spatial_dimension] && in HandleConvolution()
640 kernel_limits[spatial_dimension] == output_limits[spatial_dimension] && in HandleConvolution()
651 for (int64 kernel_idx = 0; kernel_idx < kernel_limits[spatial_dimension]; in HandleConvolution()