def get_ds_test_input_fn():
    np.random.seed(_RANDOM_SEED)
    (_, (x_test, y_test)) = testing_utils.get_test_data(train_samples=_TRAIN_SIZE, test_samples=50, input_shape=_INPUT_SIZE, num_classes=_NUM_CLASS)
    y_test = keras.utils.to_categorical(y_test)
    dataset = dataset_ops.Dataset.from_tensor_slices((x_test, y_test))
    dataset = dataset.batch(32)
    return dataset
def get_ds_test_input_fn():
    np.random.seed(_RANDOM_SEED)
    (_, (x_test, y_test)) = testing_utils.get_test_data(train_samples=_TRAIN_SIZE, test_samples=50, input_shape=_INPUT_SIZE, num_classes=_NUM_CLASS)
    y_test = keras.utils.to_categorical(y_test)
    dataset = dataset_ops.Dataset.from_tensor_slices((x_test, y_test))
    dataset = dataset.batch(32)
    return dataset