Searched refs:input_spatial_dim (Results 1 – 2 of 2) sorted by relevance
477 const int64 input_spatial_dim = in HandleConvolution() local479 const int64 input_limit = lhs_shape.dimensions(input_spatial_dim); in HandleConvolution()
1118 const int64 input_spatial_dim = dnums.input_spatial_dimensions(ki); in HandleConvolution() local1146 lhs_spatial_index < lhs_shape.dimensions(input_spatial_dim))) { in HandleConvolution()1151 lhs_spatial_index * lhs_dim_multipliers[input_spatial_dim]; in HandleConvolution()