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

Public Member Functions

TAlgorithm getAlgorithm (at::IntArrayRef tensorDimensions1, at::IntArrayRef 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: