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_


struct TensorIterator

Detailed Description

Definition at line 258 of file TensorIterator.h.

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