Searched defs:flatten_spec (Results 1 – 1 of 1) sorted by relevance
298 def __tensor_unflatten__(inner_tensors, flatten_spec, outer_size, outer_stride): argument328 def __coerce_same_metadata_as_tangent__(self, flatten_spec): argument