LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
sort.hpp File Reference
#include "lbann/layers/data_type_layer.hpp"
#include "lbann/layers/layer.hpp"
#include "lbann/proto/datatype_helpers.hpp"
#include "lbann/proto/layers.pb.h"
#include "lbann/macros/instantiate_device.hpp"
Include dependency graph for sort.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  lbann::sort_layer< TensorDataType, T_layout, Dev >
 Sort tensor entries. More...
 

Namespaces

 lbann
 

Macros

#define PROTO_DEVICE(T, Device)   extern template class sort_layer<T, data_layout::DATA_PARALLEL, Device>
 

Functions

 lbann::PROTO (float)
 
 lbann::PROTO (double)
 

Macro Definition Documentation

◆ PROTO_DEVICE

#define PROTO_DEVICE (   T,
  Device 
)    extern template class sort_layer<T, data_layout::DATA_PARALLEL, Device>

Definition at line 113 of file sort.hpp.