Loading...
Searching...
No Matches
tiny_dnn::core::OpKernelContext Member List

This is the complete list of members for tiny_dnn::core::OpKernelContext, including all inherited members.

device() const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
engine() const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
input(const int idx) const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
input_grad(const int idx) const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
Layer() const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
OpKernelContext(const std::vector< tensor_t * > &in_data, std::vector< tensor_t * > &out_data) (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline explicit
OpKernelContext(const std::vector< tensor_t * > &in_data, const std::vector< tensor_t * > &out_data, std::vector< tensor_t * > &out_grad, std::vector< tensor_t * > &in_grad) (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline explicit
output(const int idx) const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
output_grad(const int idx) const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
parallelize() const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
params() const (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
setDevice(Device *device) (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
setEngine(const backend_t engine) (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
setLayer(layer *layer) (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
setParallelize(const bool parallelize) (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline
setParams(Params *params) (defined in tiny_dnn::core::OpKernelContext) | tiny_dnn::core::OpKernelContext | inline