24.04
|
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator. More...
#include <ReshapeAttributes.h>
Public Member Functions | |
ReshapeAttributes & | shape (const TensorShape &shape) |
Set destination shape to reshape to. More... | |
TensorShape | shape () const |
Get destination shape to reshape to. More... | |
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
Reshape attributes
Definition at line 38 of file ReshapeAttributes.h.
TensorShape shape | ( | ) | const |
Get destination shape to reshape to.
Definition at line 38 of file ReshapeAttributes.cpp.
Referenced by ReshapeAttributes::shape().
ReshapeAttributes & shape | ( | const TensorShape & | shape | ) |
Set destination shape to reshape to.
Definition at line 33 of file ReshapeAttributes.cpp.
References ReshapeAttributes::shape().
Referenced by GpuReshape::create_op(), and GpuReshape::validate_op().