Searched defs:img_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/tensorboard/ |
| H A D | writer.py | 578 self, tag, img_tensor, global_step=None, walltime=None, dataformats="CHW" argument
629 self, tag, img_tensor, global_step=None, walltime=None, dataformats="NCHW" argument
675 img_tensor, argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_bundled_images.py | 39 def bundle_jpeg_image(img_tensor, quality): argument
|