Searched refs:_create_write_items_for_dtensor (Results 1 – 2 of 2) sorted by relevance
74 def _create_write_items_for_dtensor(fqn: str, tensor: DTensor) -> WriteItem: function210 requests.append(_create_write_items_for_dtensor(fqn, obj))228 return [_create_write_items_for_dtensor(fqn, object)]
592 _create_write_items_for_dtensor,598 return [_create_write_items_for_dtensor(fqn, object)]