code: def test_fuzz_nhwc(self):
      input_size_pairs = (
          ((7, 77, 7, 77), (5, 55, 5, 55)),
          ((1, 1, 777, 1), (1, 1, 555, 1)),
          ((5, 47, 5, 55), (1, 1, 1, 2048 * 32 + 1)),
          ((1, 1, 1, 2048 * 32 + 1), (55, 47, 5, 55)),
          ((555, 1, 1, 1), (32, 8, 32, 8)),
          ((32, 8, 32, 8), (55, 47, 5, 55)),
          ((1, 1, 33333, 1), (55, 47, 55, 5)),
          ((55, 47, 55, 5), (1, 1, 33333, 1)),
      )
      appliers = (
          MultiTensorApply(2048 * 32),
          MultiTensorApply(333),
          MultiTensorApply(33333),
      )
      repeat_tensors = (1, 55)
  
      for sizea, sizeb in input_size_pairs:
          for applier in appliers:
              for repeat in repeat_tensors:
                  for x_type in (torch.float32, torch.float16):
                      for y_type in (torch.float32, torch.float16):
                          for out_type in (
                              torch.float32,
                              torch.float16,
                          ):
                              for inplace in (True, False):
                                  if inplace is True and (
                                      y_type is not out_type
                                  ):
                                      continue
                                  else:
                                      self.axpby(
                                          sizea,
                                          sizeb,
                                          applier,
                                          repeat,
                                          x_type,
                                          y_type,
                                          out_type,
                                          inplace=inplace,
                                          nhwc=True,
                                      )
  
file_path: apex/tests/L0/run_amp/test_multi_tensor_axpby.py
imports: import unittest
  import functools
  import itertools
  from apex import amp
  import torch
  from torch import nn
  import torch.nn.functional
  from math import floor
  from utils import common_init, HALF, FLOAT, ALWAYS_HALF, ALWAYS_FLOAT, MATCH_INPUT
label_desc: Use image node_id 5 for calling the TestMultiTensorAxpby obj's underlying member method code with example usage: obj.test_fuzz_nhwc() without return types