Searched refs:input_dim_sizes (Results 1 – 1 of 1) sorted by relevance
37 std::vector<int64> input_dim_sizes(num_spatial_dims); in AvgPoolDivideByCountWithGeneralPadding() local45 input_dim_sizes[i] = input_shape[dim]; in AvgPoolDivideByCountWithGeneralPadding()53 auto ones = Broadcast(One(b, dtype), input_dim_sizes); in AvgPoolDivideByCountWithGeneralPadding()