Package index • pjrt

All functions

as.character(<PJRTElementType>): Convert PJRTElementType to string
as_array(<PJRTBuffer>): Convert a PJRTBuffer to an R Array
as_array_async(): Convert buffer to R array asynchronously
as_pjrt_client(): Convert to PJRT Client
as_pjrt_device(): Convert to PJRT Device
as_pjrt_plugin(): Convert to PJRT Plugin
await(): Await an async operation
build_tree(): Build Tree
copy_buffer(): Copy Buffer to Device
devices(): Devices
dispatch() dispatcher_size(): Native eager-dispatch fast path
dispatcher(): Dispatcher
elt_type(): Element Type
flatten(): Flatten
flatten_fun(): Flatten a Function
format_buffer(): Format Buffer Data
inspect_hlo(): Inspect the HLO Source of a Program
install_pjrt(): Install PJRT Plugins
is_ready(): Check if an async operation is ready
map_tree(): Map Over a Tree
pjrt pjrt-package: pjrt: R Interface to PJRT
pjrt_buffer() pjrt_scalar() pjrt_empty(): Create a PJRT Buffer
pjrt_client(): Create a Client
pjrt_compile(): Compile a Program
pjrt_device(): Create a PJRT Device
pjrt_execute(): Execute a PJRT program
pjrt_execution_options(): Create Execution Options
pjrt_plugin(): Create PJRT Plugin
pjrt_program(): Create a PJRTProgram
pjrt_register_custom_call(): Register a Custom Call Handler
platform(): Platform Name
plugin_attributes(): Get Plugin Attributes
plugin_client_create(): Create PJRT Client
plugins_downloaded(): Check if Plugin is Downloaded
pmap_tree(): Map Over Multiple Trees
print(<PJRTBuffer>): Print a PJRT Buffer
tree_child_kinds(): Top-Level Child Kinds
tree_child_names(): Top-Level Child Names
tree_child_sizes(): Top-Level Child Sizes
tree_concat(): Concatenate Trees Under a New Root
tree_diff(): Difference Between Trees
tree_equal(): Compare Tree Structures
tree_filter_by_names(): Filter a Tree by Top-Level Names
tree_hash(): Structural Tree Hash
tree_leaf_mask(): Per-Leaf Mask From Top-Level Groups
tree_path(): Tree Path
tree_repr(): Canonical Tree Representation
tree_root_kind(): Tree Root Node Kind
tree_size(): Tree Size
unflatten(): Unflatten
value(): Get the value of an async operation