Expand description
shape dialect.
This dialect contains operations for shape inference.
Note: Unless explicitly stated, all functions that return a shape and take shapes as input, return the invalid shape if one of its operands is an invalid shape. This avoids flagging multiple errors for one verification failure. The dialect itself does not specify how errors should be combined (there are multiple different options, from always choosing first operand, concatting etc. on how to combine them).
Structs§
- An
addoperation. Addition of sizes and indices. - A builder for an
addoperation. - An
anyoperation. Return any combination of the input shapes. - A builder for an
anyoperation. - An
assuming_alloperation. Return a logical AND of all witnesses. - A builder for an
assuming_alloperation. - An
assumingoperation. Execute the region. - A builder for an
assumingoperation. - An
assuming_yieldoperation. Yield operation. - A builder for an
assuming_yieldoperation. - A
broadcastoperation. Returns the broadcasted output shape of two or more inputs. - A builder for a
broadcastoperation. - A
concatoperation. Concatenates two shapes. - A builder for a
concatoperation. - A
const_shapeoperation. Creates a constant shape or extent tensor. - A builder for a
const_shapeoperation. - A
const_sizeoperation. Creates a constant of typeshape.size. - A builder for a
const_sizeoperation. - A
const_witnessoperation. An operation that returns a statically known witness value. - A builder for a
const_witnessoperation. - A
cstr_broadcastableoperation. Determines if 2+ shapes can be successfully broadcasted. - A builder for a
cstr_broadcastableoperation. - A
cstr_eqoperation. Determines if all input shapes are equal. - A builder for a
cstr_eqoperation. - A
cstr_requireoperation. Represents a runtime assertion that an i1 istrue. - A builder for a
cstr_requireoperation. - A
debug_printoperation. Prints the input shape or size. - A builder for a
debug_printoperation. - A
dimoperation. Gets the specified extent from the shape of a shaped input. - A builder for a
dimoperation. - A
divoperation. Division of sizes and indices. - A builder for a
divoperation. - A
from_extent_tensoroperation. Creates a shape from a tensor of extents. - A builder for a
from_extent_tensoroperation. - A
from_extentsoperation. Creates a shape from extents. - A builder for a
from_extentsoperation. - A
funcoperation. Shape function. - A builder for a
funcoperation. - A
function_libraryoperation. Represents shape functions and corresponding ops. - A builder for a
function_libraryoperation. - A
get_extentoperation. Gets the specified extent from a shape or extent tensor. - A builder for a
get_extentoperation. - An
index_to_sizeoperation. Converts a standard index to a shape size. - A builder for an
index_to_sizeoperation. - An
is_broadcastableoperation. Determines if 2+ shapes can be successfully broadcasted. - A builder for an
is_broadcastableoperation. - A
maxoperation. Elementwise maximum. - A builder for a
maxoperation. - A
meetoperation. Returns the least general shape or size of its operands. - A builder for a
meetoperation. - A
minoperation. Elementwise minimum. - A builder for a
minoperation. - A
muloperation. Multiplication of sizes and indices. - A builder for a
muloperation. - A
num_elementsoperation. Returns the number of elements for a given shape. - A builder for a
num_elementsoperation. - A
rankoperation. Gets the rank of a shape. - A builder for a
rankoperation. - A
reduceoperation. Returns an expression reduced over a shape or extent tensor. - A builder for a
reduceoperation. - A
returnoperation. Shape function return operation. - A builder for a
returnoperation. - A
shape_eqoperation. Returns whether the input shapes or extent tensors are equal. - A builder for a
shape_eqoperation. - A
shape_ofoperation. Returns shape of a value or shaped type operand. - A builder for a
shape_ofoperation. - A
size_to_indexoperation. Casts between index types of the shape and standard dialect. - A builder for a
size_to_indexoperation. - A
split_atoperation. Splits a shape at a given index. - A builder for a
split_atoperation. - A
to_extent_tensoroperation. Creates a dimension tensor from a shape. - A builder for a
to_extent_tensoroperation. - A
value_as_shapeoperation. Returns value as a shape. - A builder for a
value_as_shapeoperation. - A
value_ofoperation. Returns value of a !shape.value_shape operand. - A builder for a
value_ofoperation. - A
with_shapeoperation. Returns ValueShape with given shape. - A builder for a
with_shapeoperation. - A
yieldoperation. Returns the value to parent op. - A builder for a
yieldoperation.
Functions§
- Creates an
addoperation. - Creates an
anyoperation. - Creates an
assumingoperation. - Creates an
assuming_alloperation. - Creates an
assuming_yieldoperation. - Creates a
broadcastoperation. - Creates a
concatoperation. - Creates a
const_shapeoperation. - Creates a
const_sizeoperation. - Creates a
const_witnessoperation. - Creates a
cstr_broadcastableoperation. - Creates a
cstr_eqoperation. - Creates a
cstr_requireoperation. - Creates a
debug_printoperation. - Creates a
dimoperation. - Creates a
divoperation. - Creates a
from_extent_tensoroperation. - Creates a
from_extentsoperation. - Creates a
funcoperation. - Creates a
function_libraryoperation. - Creates a
get_extentoperation. - Creates an
index_to_sizeoperation. - Creates an
is_broadcastableoperation. - Creates a
maxoperation. - Creates a
meetoperation. - Creates a
minoperation. - Creates a
muloperation. - Creates a
num_elementsoperation. - Creates a
rankoperation. - Creates a
reduceoperation. - Creates a
returnoperation. - Creates a
shape_eqoperation. - Creates a
shape_ofoperation. - Creates a
size_to_indexoperation. - Creates a
split_atoperation. - Creates a
to_extent_tensoroperation. - Creates a
value_as_shapeoperation. - Creates a
value_ofoperation. - Creates a
with_shapeoperation. - Creates a
yieldoperation.