Searched refs:num_row_squares (Results 1 – 1 of 1) sorted by relevance
211 void SetCheckerboardImageInput(int batch_size, int num_row_squares, in SetCheckerboardImageInput() argument217 const int64 image_size = num_row_squares * square_size * row_size; in SetCheckerboardImageInput()223 for (int y = 0; y < num_row_squares; ++y) { in SetCheckerboardImageInput()241 TensorShape({batch_size, num_row_squares * square_size, in SetCheckerboardImageInput()