Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Protected Attributes | Friends
at::TensorIterator::Builder Struct Reference

Public Member Functions

void add_output (const Tensor &output, const Type *type=nullptr)
 
void add_input (const Tensor &input, const Type *type=nullptr)
 
void dont_compute_common_dtype ()
 
void dont_resize_outputs ()
 
std::unique_ptr< TensorIteratorbuild ()
 

Protected Attributes

std::unique_ptr< TensorIteratoriter_
 

Friends

struct TensorIterator
 

Detailed Description

Definition at line 258 of file TensorIterator.h.


The documentation for this struct was generated from the following files: