Searched refs:layout_fn (Results 1 – 2 of 2) sorted by relevance
356 layout_fn = self._create_layout_file(layout_map)359 self._target_command, layout_fn, ' -i '.join(write_list), tmpfn)371 self._remove_temp_file(layout_fn)
748 def Compile(self, argument_shapes=(), compile_options=None, layout_fn=None, argument769 if layout_fn:771 shape.map_leaves(layout_fn) for shape in argument_shapes773 result_shape = result_shape.map_leaves(layout_fn)786 layout_fn=None, argument791 layout_fn=layout_fn,