Searched defs:dataformats (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
680 dataformats="CHW", argument
|
| H A D | summary.py | 544 def image(tag, tensor, rescale=1, dataformats="NCHW"): argument
582 tag, tensor_image, tensor_boxes, rescale=1, dataformats="CHW", labels=None argument
|