View Source EXLA.MLIR.Value (EXLA v0.7.3)
Representation of an MLIR Value.
MLIR Values are SSA and generally are either operations or block arguments. This module is used to construct most of the MLIR operations.
Summary
Functions
Link to this function
convolution(tensor, kernel, strides, padding, input_dilation, kernel_dilation, dimension_numbers, feature_group_count, batch_group_count, precision_config, output_shape)
View Source
Link to this function
gather(source, indices, index_vector_dim, slice_sizes, offset_dims, collapsed_slice_dims, start_index_map)
View Source
Link to this function
scatter(target, indices, updates, kind, indices_rank, update_window_dims, inserted_window_dims, index_dims_to_window_dims)
View Source
Link to this function
select_and_scatter(target, source, init_value, comparison, window_dimensions, window_strides, padding)
View Source
Link to this function