Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::OperandInfo Struct Reference

Public Member Functions

 OperandInfo (const Tensor &t, const Type *type=nullptr)
 

Data Fields

DimVector stride_bytes
 Stride after broadcasting. The stride is in bytes, not number of elements.
 
Tensor tensor
 The original tensor operand. More...
 
Typetype = nullptr
 The desired type for the operand. More...
 
void * data = nullptr
 The data pointer. More...
 
bool is_output = false
 
bool is_read_write = false
 

Detailed Description

Definition at line 67 of file TensorIterator.h.

Field Documentation

void* at::OperandInfo::data = nullptr

The data pointer.

This may be different from tensor.data_ptr() if the iterator is split.

Definition at line 92 of file TensorIterator.h.

Tensor at::OperandInfo::tensor

The original tensor operand.

Note that the strides, data pointer, and other attributes may differ due to dimension reordering and coalescing.

Definition at line 82 of file TensorIterator.h.

Type* at::OperandInfo::type = nullptr

The desired type for the operand.

For inputs, this specifies that the input should be converted to this type if necessary. For outputs, this specifies which type to allocate. Note that there is very limited support for type conversions currently: they are only allowed for zero-dim tensors.

Definition at line 88 of file TensorIterator.h.


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