postproc_ops.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2023-2025, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
namespace ML {
namespace fil {
 
enum struct row_op : unsigned char {
  disable   = 0b00100000,
  softmax   = 0b01000000,
  max_index = 0b10000000
};
 
enum struct element_op : unsigned char {
  disable                = 0b00000000,
  signed_square          = 0b00000001,
  hinge                  = 0b00000010,
  sigmoid                = 0b00000100,
  exponential            = 0b00001000,
  logarithm_one_plus_exp = 0b00010000
};
 
}  // namespace fil
}  // namespace ML