xla 0.1.6

Bindings for the XLA C++ library.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30


xla_computation_ordered_wrappermain.8�
main.8.
Arg_0.1	parameter*
2:��0
Arg_1.2	parameter*
2:H��2

constant.3constant*:B
*B@��6
broadcast.4	broadcast*
2:����
dot.5dot*
2:��xla_computation(ordered_wrapper)/jit(main)/dot_general[dimension_numbers=(((1,), (0,)), ((), ())) precision=None preferred_element_type=None]1/home/laurent/github/jax/examples/jax_cpp/prog.py �
���
��
divide.6divide*
2:e.xla_computation(ordered_wrapper)/jit(main)/div1/home/laurent/github/jax/examples/jax_cpp/prog.py ���2
tuple.7tuple
"*
2:���"H
*
2
*
2
"*
2Arg_0Arg_1(0"H
*
2
*
2
"*
2Arg_0Arg_1(0J