Searched refs:get_style_predict_executor_shape (Results 1 – 2 of 2) sorted by relevance
35 assert style_transfer_executor.get_style_predict_executor_shape() == (1, 256, 256, 3)
93 def get_style_predict_executor_shape(self): member in StyleTransfer