Home
last modified time | relevance | path

Searched refs:map_leaves (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py567 def map_leaves(self, f): member in Shape
578 children = tuple(child.map_leaves(f) for child in self.tuple_shapes())
610 return self.map_leaves(f)
771 shape.map_leaves(layout_fn) for shape in argument_shapes
773 result_shape = result_shape.map_leaves(layout_fn)