2a4e2b1b131d9a1b48d8eba5479b585bf328d3ba,onnx_tf/backend.py,TensorflowBackend,get_data_format,#Any#Any#Any#,621

Before Change

  def get_data_format(cls, x_rank, support_cuda):
    if support_cuda:
      data_format = "NCDHW"
      if x_rank == 3:
        data_format = "NCW"
      elif x_rank == 4:
        data_format = "NCHW"
    else:
      data_format = "NDHWC"
      if x_rank == 3:
        data_format = "NWC"

After Change

    for i in range(x_rank-2):
      sp_dim_lst.append(sp_dim_names[-i-1])

    sp_dim_string = "".join(reversed(sp_dim_lst))
    storage_format = "NC" + sp_dim_string

    if support_cuda:
      compute_format = "NC" + sp_dim_string
    else:
      compute_format = "N" + sp_dim_string + "C"
    return storage_format, compute_format

Instances