Searched defs:np_tensor (Results 1 – 2 of 2) sorted by relevance
121 def _chunk_tensor(np_tensor, strategy, depth): argument150 def _chunk_tensor_by_strategy(np_tensor, strategy): argument
413 def _convert_image_format(np_tensor, input_format, out_format='HWC'): argument