Searched defs:xla_shape (Results 1 – 1 of 1) sorted by relevance
374 def _check_special(name, xla_shape, buf): argument827 def _xla_param(builder, param_num, xla_shape, replicated, partitions, parts_proto): argument973 def _array_aval_from_xla_shape(xla_shape): argument1429 def zeros(xla_shape): argument