Searched refs:output_spatial_dim (Results 1 – 2 of 2) sorted by relevance
471 const int64 output_spatial_dim = in HandleConvolution() local473 const int64 output_limit = result_shape.dimensions(output_spatial_dim); in HandleConvolution()
1119 const int64 output_spatial_dim = dnums.output_spatial_dimensions(ki); in HandleConvolution() local1125 out_index[output_spatial_dim] * window_dim.stride() - in HandleConvolution()