View Source EXLA.MLIR.Value (EXLA v0.7.2)

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

Functions

Link to this function

add(func, value1, value2)

View Source
Link to this function

atan2(func, value1, value2)

View Source
Link to this function

bitcast_convert(value, dtype)

View Source
Link to this function

bitwise_and(func, value1, value2)

View Source
Link to this function

bitwise_or(func, value1, value2)

View Source
Link to this function

bitwise_xor(func, value1, value2)

View Source
Link to this function

broadcast_in_dim(operand, output_shape, axes)

View Source
Link to this function

call(function, args, function)

View Source
Link to this function

clamp(operand, min, max)

View Source
Link to this function

concatenate(operands, dimension)

View Source
Link to this function

constant_from_binary(func, data, shape)

View Source
Link to this function

constant_r0(func, value, type)

View Source
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

count_leading_zeros(value)

View Source
Link to this function

divide(func, value1, value2)

View Source
Link to this function

dot_general(output_shape, lhs, rhs, dnums, precision_config)

View Source
Link to this function

dynamic_slice(op, starts, lengths)

View Source
Link to this function

dynamic_update_slice(operand, updates, starts)

View Source
Link to this function

equal(func, value1, value2)

View Source
Link to this function

fft(value, fft_kind, fft_length)

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

get_tuple_element(value, index)

View Source
Link to this function

greater(func, value1, value2)

View Source
Link to this function

greater_equal(func, value1, value2)

View Source
Link to this function

if_op(pred, output_shapes)

View Source
Link to this function

left_shift(func, value1, value2)

View Source
Link to this function

less(func, value1, value2)

View Source
Link to this function

less_equal(func, value1, value2)

View Source
Link to this function

map(function, inputs, dimensions)

View Source
Link to this function

max(func, value1, value2)

View Source
Link to this function

min(func, value1, value2)

View Source
Link to this function

multiply(func, value1, value2)

View Source
Link to this function

not_equal(func, value1, value2)

View Source
Link to this function

optimization_barrier(operand)

View Source
Link to this function

pad(operand, pad, padding_config)

View Source
Link to this function

pow(func, value1, value2)

View Source
Link to this function

qr(value, q_shape, r_shape)

View Source
Link to this function

reduce(function, init_values, inputs, dimensions)

View Source
Link to this function

remainder(func, value1, value2)

View Source
Link to this function

reshape(op, shape_tuple)

View Source
Link to this function

right_shift_arithmetic(func, value1, value2)

View Source
Link to this function

right_shift_logical(func, value1, value2)

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(pred, on_true, on_false)

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

slice(op, starts, limits, strides)

View Source
Link to this function

sort(value, comparator_fun, axis, stable)

View Source
Link to this function

subtract(func, value1, value2)

View Source
Link to this function

triangular_solve(a, b, left_side, lower, transform)

View Source
Link to this function

variadic_return(values, flatten_tuples? \\ false)

View Source
Link to this function

while(function, initial)

View Source
Link to this function

window_reduce(function, init_values, inputs, window_dimensions, window_strides, input_dilations, window_dilations, padding)

View Source