Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc589 DimensionVector output_limits; in HandleConvolution() local
594 output_limits.push_back(1); in HandleConvolution()
609 output_limits.push_back(output_limit); in HandleConvolution()
628 if (input_limits[spatial_dimension] == output_limits[spatial_dimension] && in HandleConvolution()
629 kernel_limits[spatial_dimension] == output_limits[spatial_dimension] && in HandleConvolution()
640 kernel_limits[spatial_dimension] == output_limits[spatial_dimension] && in HandleConvolution()
643 window_dim.padding_high() == output_limits[spatial_dimension] - 1 && in HandleConvolution()
644 window_dim.padding_low() == output_limits[spatial_dimension] - 1) { in HandleConvolution()
645 valid_position_counts.push_back(output_limits[spatial_dimension]); in HandleConvolution()
654 for (int64 output_idx = 0; output_idx < output_limits[spatial_dimension]; in HandleConvolution()