tfaot Namespace Reference

Classes
class	BatchRule
class	BatchStrategy
class	Model
class	Wrapper

Typedefs
typedef tensorflow::XlaCompiledCpuFunction::AllocMode	AllocMode
typedef std::vector< std::vector< bool > >	BoolArrays
typedef std::vector< std::vector< double > >	DoubleArrays
typedef std::vector< std::vector< float > >	FloatArrays
typedef std::vector< std::vector< int32_t > >	Int32Arrays
typedef std::vector< std::vector< int64_t > >	Int64Arrays

Functions
template<size_t... Index>
auto	createIndexLooper (std::index_sequence< Index... >)
template<size_t N>
auto	createIndexLooper ()
std::ostream &	operator<< (std::ostream &out, const BatchRule &rule)

Typedef Documentation

AllocMode

typedef tensorflow::XlaCompiledCpuFunction::AllocMode tfaot::AllocMode

Definition at line 20 of file Util.h.

BoolArrays

typedef std::vector<std::vector<bool> > tfaot::BoolArrays

Definition at line 21 of file Util.h.

DoubleArrays

typedef std::vector<std::vector<double> > tfaot::DoubleArrays

Definition at line 25 of file Util.h.

FloatArrays

typedef std::vector<std::vector<float> > tfaot::FloatArrays

Definition at line 24 of file Util.h.

Int32Arrays

typedef std::vector<std::vector<int32_t> > tfaot::Int32Arrays

Definition at line 22 of file Util.h.

Int64Arrays

typedef std::vector<std::vector<int64_t> > tfaot::Int64Arrays

Definition at line 23 of file Util.h.

Function Documentation

createIndexLooper() [1/2]

template<size_t... Index>

auto tfaot::createIndexLooper

(

std::index_sequence< Index... >

)

Definition at line 29 of file Util.h.

References f.

Referenced by createIndexLooper(), and tfaot::Model< W >::run().

                                                      {
    return [](auto&& f) { (f(std::integral_constant<size_t, Index>{}), ...); };
  }

createIndexLooper() [2/2]

template<size_t N>

auto tfaot::createIndexLooper

(

)

Definition at line 35 of file Util.h.

References createIndexLooper().

                           {
    return createIndexLooper(std::make_index_sequence<N>{});
  }

operator<<()

std::ostream & tfaot::operator<<	(	std::ostream &	out,
		const BatchRule &	rule
	)

Definition at line 56 of file Batching.cc.

References tfaot::BatchRule::getBatchSize(), tfaot::BatchRule::getLastPadding(), tfaot::BatchRule::getSizes(), mps_fire::i, tfaot::BatchRule::nSizes(), and MillePedeFileConverter_cfg::out.

                                                                 {
    out << "BatchRule(batchSize=" << rule.getBatchSize() << ", sizes=";
    for (size_t i = 0; i < rule.nSizes(); i++) {
      out << (i == 0 ? "" : ",") << rule.getSizes()[i];
    }
    return out << ", lastPadding=" << rule.getLastPadding() + ")";
  }