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