TraceableTensorShape Struct Reference

Traceable tensor shape. More...

Detailed Description

Traceable tensor shape.

A tensor shape can be either a concrete shape or a traceable tensor. This is useful when we need to trace a function graph and let it generalize to other batch shapes.

#include <TraceableTensorShape.h>

Inheritance diagram for TraceableTensorShape:

Public Types
using	Size = int64_t

Public Member Functions
	TraceableTensorShape (const TensorShape &shape)
	TraceableTensorShape (TensorShapeRef shape)
	TraceableTensorShape (Size shape)
	TraceableTensorShape (const ATensor &shape)
TraceableTensorShape	slice (std::size_t N, std::size_t M) const
	Slice the shape.
TraceableTensorShape	slice (std::size_t N) const
	Chop-off the first N elements of the shape, semantically the same as ArrayRef::slice, but traceable.
TensorShape	concrete () const
ATensor	as_tensor () const

Member Typedef Documentation

Size

using Size = int64_t

Constructor & Destructor Documentation

TraceableTensorShape() [1/4]

TraceableTensorShape

(

const TensorShape &

shape

)

TraceableTensorShape() [2/4]

TraceableTensorShape

(

TensorShapeRef

shape

)

TraceableTensorShape() [3/4]

TraceableTensorShape

(

Size

shape

)

TraceableTensorShape() [4/4]

TraceableTensorShape

(

const ATensor &

shape

)

Member Function Documentation

as_tensor()

ATensor as_tensor

(

)

const

Returns

the shape represented as a scalar tensor (possibly traceable)

concrete()

TensorShape concrete

(

)

const

Returns

the concrete shape (without any traceable information)

slice() [1/2]

TraceableTensorShape slice

(

std::size_t

N

)

const

Chop-off the first N elements of the shape, semantically the same as ArrayRef::slice, but traceable.

slice() [2/2]

TraceableTensorShape slice	(	std::size_t	N,
		std::size_t	M ) const

Slice the shape.