Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::AlgorithmsCache< TAlgorithm > Class Template Reference

Public Member Functions

TAlgorithm getAlgorithm (const std::vector< TIndex > &tensorDimensions1, const std::vector< TIndex > &tensorDimensions2, int algorithmFlags, std::function< TAlgorithm()> generatingFunc)
 

Detailed Description

template<typename TAlgorithm>
class caffe2::AlgorithmsCache< TAlgorithm >

Definition at line 13 of file conv_op_cache_cudnn.h.


The documentation for this class was generated from the following file: