Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc505 const int64 lhs_spatial_index = in HandleConvolution() local
511 if (undilated_index != lhs_spatial_index * window_dim.base_dilation()) { in HandleConvolution()
516 if (lhs_spatial_index < 0 || in HandleConvolution()
517 lhs_spatial_index >= input_limits[spatial_dimension]) { in HandleConvolution()
Dhlo_evaluator_typed_visitor.h1137 int64 lhs_spatial_index; in HandleConvolution() local
1139 lhs_spatial_index = undilated_index / window_dim.base_dilation(); in HandleConvolution()
1141 lhs_spatial_index = undilated_index; in HandleConvolution()
1145 if (!(lhs_spatial_index >= 0 && in HandleConvolution()
1146 lhs_spatial_index < lhs_shape.dimensions(input_spatial_dim))) { in HandleConvolution()
1151 lhs_spatial_index * lhs_dim_multipliers[input_spatial_dim]; in HandleConvolution()