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...
Type *	type = 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:

• aten/src/ATen/native/TensorIterator.h