Hugging Face's logo

amd
/
FLUX.2-klein-4B-amdnpu

Text-to-Image
ONNX
Safetensors
English
flux
image-to-image
RyzenAI
Quantization
ONNX
Computer Vision
Model card Files
xet
 Community
FLUX.2-klein-4B-amdnpu / vae_decoder /dynamic /dd /cache /NhwcConv_0-post_quant_convConv_meta.json
huisu's picture
huisu
Upload folder using huggingface_hub
9272c13 verified
Raw
History Blame Contribute Delete
255 kB
{
"dd_meta_major_version": 1,
"dd_meta_minor_version": 4,
"state_table_updates": [],
"op_list": [
{
"name": "NhwcConv_0-/post_quant_conv/Conv",
"type": "SDConv",
"in_args": [
"Transpose_0_out-latents.out2_3_0"
],
"const_args": [
"NhwcConv_0_weight_NHWC"
],
"out_args": [
"NhwcConv_0_out-/post_quant_conv/Conv_output_0.out2_3_0"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"1",
"1"
]
},
"pads": {
"type": "int",
"value": [
"0",
"0",
"0",
"0"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"32"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"32"
]
},
"weight_shape": {
"type": "int",
"value": [
"32",
"1",
"1",
"32"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_1-/decoder/conv_in/Conv",
"type": "SDConv",
"in_args": [
"NhwcConv_0_out-/post_quant_conv/Conv_output_0.out2_3_0"
],
"const_args": [
"NhwcConv_1_weight_NHWC"
],
"out_args": [
"NhwcConv_1_out-/decoder/conv_in/Conv_output_0.out2_3_1"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"32"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"32"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_0",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_1_out-/decoder/conv_in/Conv_output_0.out2_3_1"
],
"const_args": [
"GroupNorm_0_wts_10_0_0"
],
"out_args": [
"GroupNorm_0.out14_0.out10_0_0"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_2-/decoder/mid_block/resnets.0/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_0.out14_0.out10_0_0"
],
"const_args": [
"NhwcConv_2_weight_NHWC"
],
"out_args": [
"NhwcConv_2_out-/decoder/mid_block/resnets.0/conv1/Conv_output_0.out2_3_2"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_1",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_2_out-/decoder/mid_block/resnets.0/conv1/Conv_output_0.out2_3_2"
],
"const_args": [
"GroupNorm_1_wts_10_0_1"
],
"out_args": [
"GroupNorm_1.out14_1.out10_0_1"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_3-/decoder/mid_block/resnets.0/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_1.out14_1.out10_0_1"
],
"const_args": [
"NhwcConv_3_weight_NHWC"
],
"out_args": [
"NhwcConv_3_out-/decoder/mid_block/resnets.0/conv2/Conv_output_0.out2_3_3"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/mid_block/resnets.0/Add",
"type": "SDAdd",
"in_args": [
"NhwcConv_1_out-/decoder/conv_in/Conv_output_0.out2_3_1",
"NhwcConv_3_out-/decoder/mid_block/resnets.0/conv2/Conv_output_0.out2_3_3"
],
"const_args": [],
"out_args": [
"/decoder/mid_block/resnets.0/Add.out_7_1_0"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_2",
"type": "SDGroupNorm",
"in_args": [
"/decoder/mid_block/resnets.0/Add.out_7_1_0"
],
"const_args": [
"GroupNorm_2_wts_10_1_0"
],
"out_args": [
"/decoder/mid_block/attentions.0/group_norm/Add_output_0_4d_NHWC.out10_1_0"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/mid_block/attentions.0/to_k/MatMul",
"type": "SDGemm",
"in_args": [
"/decoder/mid_block/attentions.0/group_norm/Add_output_0_4d_NHWC.out10_1_0"
],
"const_args": [
"onnx::MatMul_920_15_0_0"
],
"out_args": [
"/decoder/mid_block/attentions.0/to_k/Add_output_0.out3_5_0"
],
"attrs": {
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"bias_enable": {
"type": "int",
"value": [
"1"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"bfloat16"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/mid_block/attentions.0/to_q/MatMul",
"type": "SDGemm",
"in_args": [
"/decoder/mid_block/attentions.0/group_norm/Add_output_0_4d_NHWC.out10_1_0"
],
"const_args": [
"onnx::MatMul_919_15_0_1"
],
"out_args": [
"/decoder/mid_block/attentions.0/to_q/Add_output_0.out3_5_1"
],
"attrs": {
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"bias_enable": {
"type": "int",
"value": [
"1"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"bfloat16"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/mid_block/attentions.0/to_v/MatMul",
"type": "SDGemm",
"in_args": [
"/decoder/mid_block/attentions.0/group_norm/Add_output_0_4d_NHWC.out10_1_0"
],
"const_args": [
"onnx::MatMul_921_15_0_2"
],
"out_args": [
"/decoder/mid_block/attentions.0/to_v/Add_output_0.out3_5_2"
],
"attrs": {
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"bias_enable": {
"type": "int",
"value": [
"1"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"bfloat16"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/mid_block/attentions.0/MatMulmha_5_0_0",
"type": "SDMHA_VAE",
"in_args": [
"/decoder/mid_block/attentions.0/to_q/Add_output_0.out3_5_1",
"/decoder/mid_block/attentions.0/to_k/Add_output_0.out3_5_0",
"/decoder/mid_block/attentions.0/to_v/Add_output_0.out3_5_2"
],
"const_args": [
"/decoder/mid_block/attentions.0/MatMulmha_5_0_0_mask.9_0"
],
"out_args": [
"/decoder/mid_block/attentions.0/Reshape_4_output_0.out9_0"
],
"attrs": {
"num_heads": {
"type": "int",
"value": [
"1"
]
},
"unidirectional": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512",
"latent_h*latent_w"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"op_version": {
"type": "str",
"value": [
"v1.0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/mid_block/attentions.0/to_out.0/MatMul",
"type": "SDGemm",
"in_args": [
"/decoder/mid_block/attentions.0/Reshape_4_output_0.out9_0"
],
"const_args": [
"onnx::MatMul_929_15_0_3"
],
"out_args": [
"/decoder/mid_block/attentions.0/Transpose_6_output_0_4d.out3_1_0"
],
"attrs": {
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h*latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"bias_enable": {
"type": "int",
"value": [
"1"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"bfloat16"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/mid_block/attentions.0/Add",
"type": "SDAdd",
"in_args": [
"/decoder/mid_block/attentions.0/Transpose_6_output_0_4d.out3_1_0",
"/decoder/mid_block/resnets.0/Add.out_7_1_0"
],
"const_args": [],
"out_args": [
"/decoder/mid_block/attentions.0/Add_output_0_NHWC.out_7_1_1"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_3",
"type": "SDGroupNorm",
"in_args": [
"/decoder/mid_block/attentions.0/Add_output_0_NHWC.out_7_1_1"
],
"const_args": [
"GroupNorm_3_wts_10_0_2"
],
"out_args": [
"GroupNorm_3.out14_3.out10_0_2"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_4-/decoder/mid_block/resnets.1/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_3.out14_3.out10_0_2"
],
"const_args": [
"NhwcConv_4_weight_NHWC"
],
"out_args": [
"NhwcConv_4_out-/decoder/mid_block/resnets.1/conv1/Conv_output_0.out2_3_4"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_4",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_4_out-/decoder/mid_block/resnets.1/conv1/Conv_output_0.out2_3_4"
],
"const_args": [
"GroupNorm_4_wts_10_0_3"
],
"out_args": [
"GroupNorm_4.out14_4.out10_0_3"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_5-/decoder/mid_block/resnets.1/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_4.out14_4.out10_0_3"
],
"const_args": [
"NhwcConv_5_weight_NHWC"
],
"out_args": [
"NhwcConv_5_out-/decoder/mid_block/resnets.1/conv2/Conv_output_0.out2_3_5"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/mid_block/resnets.1/Add",
"type": "SDAdd",
"in_args": [
"/decoder/mid_block/attentions.0/Add_output_0_NHWC.out_7_1_1",
"NhwcConv_5_out-/decoder/mid_block/resnets.1/conv2/Conv_output_0.out2_3_5"
],
"const_args": [],
"out_args": [
"/decoder/mid_block/resnets.1/Add_output_0_NHWC.out_7_1_2"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_5",
"type": "SDGroupNorm",
"in_args": [
"/decoder/mid_block/resnets.1/Add_output_0_NHWC.out_7_1_2"
],
"const_args": [
"GroupNorm_5_wts_10_0_4"
],
"out_args": [
"GroupNorm_5.out14_5.out10_0_4"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_6-/decoder/up_blocks.0/resnets.0/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_5.out14_5.out10_0_4"
],
"const_args": [
"NhwcConv_6_weight_NHWC"
],
"out_args": [
"NhwcConv_6_out-/decoder/up_blocks.0/resnets.0/conv1/Conv_output_0.out2_3_6"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_6",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_6_out-/decoder/up_blocks.0/resnets.0/conv1/Conv_output_0.out2_3_6"
],
"const_args": [
"GroupNorm_6_wts_10_0_5"
],
"out_args": [
"GroupNorm_6.out14_6.out10_0_5"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_7-/decoder/up_blocks.0/resnets.0/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_6.out14_6.out10_0_5"
],
"const_args": [
"NhwcConv_7_weight_NHWC"
],
"out_args": [
"NhwcConv_7_out-/decoder/up_blocks.0/resnets.0/conv2/Conv_output_0.out2_3_7"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.0/resnets.0/Add",
"type": "SDAdd",
"in_args": [
"/decoder/mid_block/resnets.1/Add_output_0_NHWC.out_7_1_2",
"NhwcConv_7_out-/decoder/up_blocks.0/resnets.0/conv2/Conv_output_0.out2_3_7"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.0/resnets.0/Add_output_0_NHWC.out_7_1_3"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_7",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.0/resnets.0/Add_output_0_NHWC.out_7_1_3"
],
"const_args": [
"GroupNorm_7_wts_10_0_6"
],
"out_args": [
"GroupNorm_7.out14_7.out10_0_6"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_8-/decoder/up_blocks.0/resnets.1/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_7.out14_7.out10_0_6"
],
"const_args": [
"NhwcConv_8_weight_NHWC"
],
"out_args": [
"NhwcConv_8_out-/decoder/up_blocks.0/resnets.1/conv1/Conv_output_0.out2_3_8"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_8",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_8_out-/decoder/up_blocks.0/resnets.1/conv1/Conv_output_0.out2_3_8"
],
"const_args": [
"GroupNorm_8_wts_10_0_7"
],
"out_args": [
"GroupNorm_8.out14_8.out10_0_7"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_9-/decoder/up_blocks.0/resnets.1/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_8.out14_8.out10_0_7"
],
"const_args": [
"NhwcConv_9_weight_NHWC"
],
"out_args": [
"NhwcConv_9_out-/decoder/up_blocks.0/resnets.1/conv2/Conv_output_0.out2_3_9"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.0/resnets.1/Add",
"type": "SDAdd",
"in_args": [
"/decoder/up_blocks.0/resnets.0/Add_output_0_NHWC.out_7_1_3",
"NhwcConv_9_out-/decoder/up_blocks.0/resnets.1/conv2/Conv_output_0.out2_3_9"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.0/resnets.1/Add_output_0_NHWC.out_7_1_4"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_9",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.0/resnets.1/Add_output_0_NHWC.out_7_1_4"
],
"const_args": [
"GroupNorm_9_wts_10_0_8"
],
"out_args": [
"GroupNorm_9.out14_9.out10_0_8"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_10-/decoder/up_blocks.0/resnets.2/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_9.out14_9.out10_0_8"
],
"const_args": [
"NhwcConv_10_weight_NHWC"
],
"out_args": [
"NhwcConv_10_out-/decoder/up_blocks.0/resnets.2/conv1/Conv_output_0.out2_3_10"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_10",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_10_out-/decoder/up_blocks.0/resnets.2/conv1/Conv_output_0.out2_3_10"
],
"const_args": [
"GroupNorm_10_wts_10_0_9"
],
"out_args": [
"GroupNorm_10.out14_10.out10_0_9"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_11-/decoder/up_blocks.0/resnets.2/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_10.out14_10.out10_0_9"
],
"const_args": [
"NhwcConv_11_weight_NHWC"
],
"out_args": [
"NhwcConv_11_out-/decoder/up_blocks.0/resnets.2/conv2/Conv_output_0.out2_3_11"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.0/resnets.2/Add",
"type": "SDAdd",
"in_args": [
"/decoder/up_blocks.0/resnets.1/Add_output_0_NHWC.out_7_1_4",
"NhwcConv_11_out-/decoder/up_blocks.0/resnets.2/conv2/Conv_output_0.out2_3_11"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.0/resnets.2/Add_output_0.nhwc7_0.out_7_1_5"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.0/upsamplers.0/Resize",
"type": "SDResize",
"in_args": [
"/decoder/up_blocks.0/resnets.2/Add_output_0.nhwc7_0.out_7_1_5"
],
"const_args": [
"/decoder/up_blocks.0/upsamplers.0/Resize.weights6_0"
],
"out_args": [
"/decoder/up_blocks.0/upsamplers.0/Resize_output_0.nhwc7_0.out_6_0"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"latent_h",
"latent_w",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_12-/decoder/up_blocks.0/upsamplers.0/conv/Conv",
"type": "SDConv",
"in_args": [
"/decoder/up_blocks.0/upsamplers.0/Resize_output_0.nhwc7_0.out_6_0"
],
"const_args": [
"NhwcConv_12_weight_NHWC"
],
"out_args": [
"NhwcConv_12_out-/decoder/up_blocks.0/upsamplers.0/conv/Conv_output_0.out2_3_12"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_11",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_12_out-/decoder/up_blocks.0/upsamplers.0/conv/Conv_output_0.out2_3_12"
],
"const_args": [
"GroupNorm_11_wts_10_0_10"
],
"out_args": [
"GroupNorm_11.out14_11.out10_0_10"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_13-/decoder/up_blocks.1/resnets.0/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_11.out14_11.out10_0_10"
],
"const_args": [
"NhwcConv_13_weight_NHWC"
],
"out_args": [
"NhwcConv_13_out-/decoder/up_blocks.1/resnets.0/conv1/Conv_output_0.out2_3_13"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_12",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_13_out-/decoder/up_blocks.1/resnets.0/conv1/Conv_output_0.out2_3_13"
],
"const_args": [
"GroupNorm_12_wts_10_0_11"
],
"out_args": [
"GroupNorm_12.out14_12.out10_0_11"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_14-/decoder/up_blocks.1/resnets.0/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_12.out14_12.out10_0_11"
],
"const_args": [
"NhwcConv_14_weight_NHWC"
],
"out_args": [
"NhwcConv_14_out-/decoder/up_blocks.1/resnets.0/conv2/Conv_output_0.out2_3_14"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.1/resnets.0/Add",
"type": "SDAdd",
"in_args": [
"NhwcConv_12_out-/decoder/up_blocks.0/upsamplers.0/conv/Conv_output_0.out2_3_12",
"NhwcConv_14_out-/decoder/up_blocks.1/resnets.0/conv2/Conv_output_0.out2_3_14"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.1/resnets.0/Add.out_7_1_6"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_13",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.1/resnets.0/Add.out_7_1_6"
],
"const_args": [
"GroupNorm_13_wts_10_0_12"
],
"out_args": [
"GroupNorm_13.out14_13.out10_0_12"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_15-/decoder/up_blocks.1/resnets.1/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_13.out14_13.out10_0_12"
],
"const_args": [
"NhwcConv_15_weight_NHWC"
],
"out_args": [
"NhwcConv_15_out-/decoder/up_blocks.1/resnets.1/conv1/Conv_output_0.out2_3_15"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_14",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_15_out-/decoder/up_blocks.1/resnets.1/conv1/Conv_output_0.out2_3_15"
],
"const_args": [
"GroupNorm_14_wts_10_0_13"
],
"out_args": [
"GroupNorm_14.out14_14.out10_0_13"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_16-/decoder/up_blocks.1/resnets.1/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_14.out14_14.out10_0_13"
],
"const_args": [
"NhwcConv_16_weight_NHWC"
],
"out_args": [
"NhwcConv_16_out-/decoder/up_blocks.1/resnets.1/conv2/Conv_output_0.out2_3_16"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.1/resnets.1/Add",
"type": "SDAdd",
"in_args": [
"/decoder/up_blocks.1/resnets.0/Add.out_7_1_6",
"NhwcConv_16_out-/decoder/up_blocks.1/resnets.1/conv2/Conv_output_0.out2_3_16"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.1/resnets.1/Add.out_7_1_7"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_15",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.1/resnets.1/Add.out_7_1_7"
],
"const_args": [
"GroupNorm_15_wts_10_0_14"
],
"out_args": [
"GroupNorm_15.out14_15.out10_0_14"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_17-/decoder/up_blocks.1/resnets.2/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_15.out14_15.out10_0_14"
],
"const_args": [
"NhwcConv_17_weight_NHWC"
],
"out_args": [
"NhwcConv_17_out-/decoder/up_blocks.1/resnets.2/conv1/Conv_output_0.out2_3_17"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_16",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_17_out-/decoder/up_blocks.1/resnets.2/conv1/Conv_output_0.out2_3_17"
],
"const_args": [
"GroupNorm_16_wts_10_0_15"
],
"out_args": [
"GroupNorm_16.out14_16.out10_0_15"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_18-/decoder/up_blocks.1/resnets.2/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_16.out14_16.out10_0_15"
],
"const_args": [
"NhwcConv_18_weight_NHWC"
],
"out_args": [
"NhwcConv_18_out-/decoder/up_blocks.1/resnets.2/conv2/Conv_output_0.out2_3_18"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.1/resnets.2/Add",
"type": "SDAdd",
"in_args": [
"/decoder/up_blocks.1/resnets.1/Add.out_7_1_7",
"NhwcConv_18_out-/decoder/up_blocks.1/resnets.2/conv2/Conv_output_0.out2_3_18"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.1/resnets.2/Add.out_7_1_8"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.1/upsamplers.0/Resize",
"type": "SDResize",
"in_args": [
"/decoder/up_blocks.1/resnets.2/Add.out_7_1_8"
],
"const_args": [
"/decoder/up_blocks.1/upsamplers.0/Resize.weights6_1"
],
"out_args": [
"/decoder/up_blocks.1/upsamplers.0/Resize_output_0.nhwc7_1.out_6_1"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"512"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_19-/decoder/up_blocks.1/upsamplers.0/conv/Conv",
"type": "SDConv",
"in_args": [
"/decoder/up_blocks.1/upsamplers.0/Resize_output_0.nhwc7_1.out_6_1"
],
"const_args": [
"NhwcConv_19_weight_NHWC"
],
"out_args": [
"NhwcConv_19_out-/decoder/up_blocks.1/upsamplers.0/conv/Conv_output_0.out2_3_19"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"512"
]
},
"weight_shape": {
"type": "int",
"value": [
"512",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_17",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_19_out-/decoder/up_blocks.1/upsamplers.0/conv/Conv_output_0.out2_3_19"
],
"const_args": [
"GroupNorm_17_wts_10_0_16"
],
"out_args": [
"GroupNorm_17.out14_17.out10_0_16"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"512"
]
},
"wts_shape": {
"type": "int",
"value": [
"1024"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"512"
]
},
"beta_shape": {
"type": "int",
"value": [
"512"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_20-/decoder/up_blocks.2/resnets.0/conv_shortcut/Conv",
"type": "SDConv",
"in_args": [
"NhwcConv_19_out-/decoder/up_blocks.1/upsamplers.0/conv/Conv_output_0.out2_3_19"
],
"const_args": [
"NhwcConv_20_weight_NHWC"
],
"out_args": [
"NhwcConv_20_out-/decoder/up_blocks.2/resnets.0/conv_shortcut/Conv_output_0.out2_3_20"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"1",
"1"
]
},
"pads": {
"type": "int",
"value": [
"0",
"0",
"0",
"0"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"weight_shape": {
"type": "int",
"value": [
"256",
"1",
"1",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_21-/decoder/up_blocks.2/resnets.0/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_17.out14_17.out10_0_16"
],
"const_args": [
"NhwcConv_21_weight_NHWC"
],
"out_args": [
"NhwcConv_21_out-/decoder/up_blocks.2/resnets.0/conv1/Conv_output_0.out2_3_21"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"512"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"weight_shape": {
"type": "int",
"value": [
"256",
"3",
"3",
"512"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_18",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_21_out-/decoder/up_blocks.2/resnets.0/conv1/Conv_output_0.out2_3_21"
],
"const_args": [
"GroupNorm_18_wts_10_0_17"
],
"out_args": [
"GroupNorm_18.out14_18.out10_0_17"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"wts_shape": {
"type": "int",
"value": [
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"256"
]
},
"beta_shape": {
"type": "int",
"value": [
"256"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_22-/decoder/up_blocks.2/resnets.0/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_18.out14_18.out10_0_17"
],
"const_args": [
"NhwcConv_22_weight_NHWC"
],
"out_args": [
"NhwcConv_22_out-/decoder/up_blocks.2/resnets.0/conv2/Conv_output_0.out2_3_22"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"weight_shape": {
"type": "int",
"value": [
"256",
"3",
"3",
"256"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.2/resnets.0/Add",
"type": "SDAdd",
"in_args": [
"NhwcConv_20_out-/decoder/up_blocks.2/resnets.0/conv_shortcut/Conv_output_0.out2_3_20",
"NhwcConv_22_out-/decoder/up_blocks.2/resnets.0/conv2/Conv_output_0.out2_3_22"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.2/resnets.0/Add.out_7_1_9"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_19",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.2/resnets.0/Add.out_7_1_9"
],
"const_args": [
"GroupNorm_19_wts_10_0_18"
],
"out_args": [
"GroupNorm_19.out14_19.out10_0_18"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"wts_shape": {
"type": "int",
"value": [
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"256"
]
},
"beta_shape": {
"type": "int",
"value": [
"256"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_23-/decoder/up_blocks.2/resnets.1/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_19.out14_19.out10_0_18"
],
"const_args": [
"NhwcConv_23_weight_NHWC"
],
"out_args": [
"NhwcConv_23_out-/decoder/up_blocks.2/resnets.1/conv1/Conv_output_0.out2_3_23"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"weight_shape": {
"type": "int",
"value": [
"256",
"3",
"3",
"256"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_20",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_23_out-/decoder/up_blocks.2/resnets.1/conv1/Conv_output_0.out2_3_23"
],
"const_args": [
"GroupNorm_20_wts_10_0_19"
],
"out_args": [
"GroupNorm_20.out14_20.out10_0_19"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"wts_shape": {
"type": "int",
"value": [
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"256"
]
},
"beta_shape": {
"type": "int",
"value": [
"256"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_24-/decoder/up_blocks.2/resnets.1/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_20.out14_20.out10_0_19"
],
"const_args": [
"NhwcConv_24_weight_NHWC"
],
"out_args": [
"NhwcConv_24_out-/decoder/up_blocks.2/resnets.1/conv2/Conv_output_0.out2_3_24"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"weight_shape": {
"type": "int",
"value": [
"256",
"3",
"3",
"256"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.2/resnets.1/Add",
"type": "SDAdd",
"in_args": [
"/decoder/up_blocks.2/resnets.0/Add.out_7_1_9",
"NhwcConv_24_out-/decoder/up_blocks.2/resnets.1/conv2/Conv_output_0.out2_3_24"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.2/resnets.1/Add.out_7_1_10"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_21",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.2/resnets.1/Add.out_7_1_10"
],
"const_args": [
"GroupNorm_21_wts_10_0_20"
],
"out_args": [
"GroupNorm_21.out14_21.out10_0_20"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"wts_shape": {
"type": "int",
"value": [
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"256"
]
},
"beta_shape": {
"type": "int",
"value": [
"256"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_25-/decoder/up_blocks.2/resnets.2/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_21.out14_21.out10_0_20"
],
"const_args": [
"NhwcConv_25_weight_NHWC"
],
"out_args": [
"NhwcConv_25_out-/decoder/up_blocks.2/resnets.2/conv1/Conv_output_0.out2_3_25"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"weight_shape": {
"type": "int",
"value": [
"256",
"3",
"3",
"256"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_22",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_25_out-/decoder/up_blocks.2/resnets.2/conv1/Conv_output_0.out2_3_25"
],
"const_args": [
"GroupNorm_22_wts_10_0_21"
],
"out_args": [
"GroupNorm_22.out14_22.out10_0_21"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"wts_shape": {
"type": "int",
"value": [
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"256"
]
},
"beta_shape": {
"type": "int",
"value": [
"256"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_26-/decoder/up_blocks.2/resnets.2/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_22.out14_22.out10_0_21"
],
"const_args": [
"NhwcConv_26_weight_NHWC"
],
"out_args": [
"NhwcConv_26_out-/decoder/up_blocks.2/resnets.2/conv2/Conv_output_0.out2_3_26"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"weight_shape": {
"type": "int",
"value": [
"256",
"3",
"3",
"256"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.2/resnets.2/Add",
"type": "SDAdd",
"in_args": [
"/decoder/up_blocks.2/resnets.1/Add.out_7_1_10",
"NhwcConv_26_out-/decoder/up_blocks.2/resnets.2/conv2/Conv_output_0.out2_3_26"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.2/resnets.2/Add.out_7_1_11"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.2/upsamplers.0/Resize",
"type": "SDResize",
"in_args": [
"/decoder/up_blocks.2/resnets.2/Add.out_7_1_11"
],
"const_args": [
"/decoder/up_blocks.2/upsamplers.0/Resize.weights6_2"
],
"out_args": [
"/decoder/up_blocks.2/upsamplers.0/Resize_output_0.nhwc7_2.out_6_2"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"256"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_27-/decoder/up_blocks.2/upsamplers.0/conv/Conv",
"type": "SDConv",
"in_args": [
"/decoder/up_blocks.2/upsamplers.0/Resize_output_0.nhwc7_2.out_6_2"
],
"const_args": [
"NhwcConv_27_weight_NHWC"
],
"out_args": [
"NhwcConv_27_out-/decoder/up_blocks.2/upsamplers.0/conv/Conv_output_0.out2_3_27"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"256"
]
},
"weight_shape": {
"type": "int",
"value": [
"256",
"3",
"3",
"256"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_23",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_27_out-/decoder/up_blocks.2/upsamplers.0/conv/Conv_output_0.out2_3_27"
],
"const_args": [
"GroupNorm_23_wts_10_0_22"
],
"out_args": [
"GroupNorm_23.out14_23.out10_0_22"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"256"
]
},
"wts_shape": {
"type": "int",
"value": [
"512"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"256"
]
},
"beta_shape": {
"type": "int",
"value": [
"256"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_28-/decoder/up_blocks.3/resnets.0/conv_shortcut/Conv",
"type": "SDConv",
"in_args": [
"NhwcConv_27_out-/decoder/up_blocks.2/upsamplers.0/conv/Conv_output_0.out2_3_27"
],
"const_args": [
"NhwcConv_28_weight_NHWC"
],
"out_args": [
"NhwcConv_28_out-/decoder/up_blocks.3/resnets.0/conv_shortcut/Conv_output_0.out2_3_28"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"1",
"1"
]
},
"pads": {
"type": "int",
"value": [
"0",
"0",
"0",
"0"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"weight_shape": {
"type": "int",
"value": [
"128",
"1",
"1",
"256"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_29-/decoder/up_blocks.3/resnets.0/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_23.out14_23.out10_0_22"
],
"const_args": [
"NhwcConv_29_weight_NHWC"
],
"out_args": [
"NhwcConv_29_out-/decoder/up_blocks.3/resnets.0/conv1/Conv_output_0.out2_3_29"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"256"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"weight_shape": {
"type": "int",
"value": [
"128",
"3",
"3",
"256"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_24",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_29_out-/decoder/up_blocks.3/resnets.0/conv1/Conv_output_0.out2_3_29"
],
"const_args": [
"GroupNorm_24_wts_10_0_23"
],
"out_args": [
"GroupNorm_24.out14_24.out10_0_23"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"wts_shape": {
"type": "int",
"value": [
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"128"
]
},
"beta_shape": {
"type": "int",
"value": [
"128"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_30-/decoder/up_blocks.3/resnets.0/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_24.out14_24.out10_0_23"
],
"const_args": [
"NhwcConv_30_weight_NHWC"
],
"out_args": [
"NhwcConv_30_out-/decoder/up_blocks.3/resnets.0/conv2/Conv_output_0.out2_3_30"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"weight_shape": {
"type": "int",
"value": [
"128",
"3",
"3",
"128"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.3/resnets.0/Add",
"type": "SDAdd",
"in_args": [
"NhwcConv_28_out-/decoder/up_blocks.3/resnets.0/conv_shortcut/Conv_output_0.out2_3_28",
"NhwcConv_30_out-/decoder/up_blocks.3/resnets.0/conv2/Conv_output_0.out2_3_30"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.3/resnets.0/Add.out_7_1_12"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_25",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.3/resnets.0/Add.out_7_1_12"
],
"const_args": [
"GroupNorm_25_wts_10_0_24"
],
"out_args": [
"GroupNorm_25.out14_25.out10_0_24"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"wts_shape": {
"type": "int",
"value": [
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"128"
]
},
"beta_shape": {
"type": "int",
"value": [
"128"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_31-/decoder/up_blocks.3/resnets.1/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_25.out14_25.out10_0_24"
],
"const_args": [
"NhwcConv_31_weight_NHWC"
],
"out_args": [
"NhwcConv_31_out-/decoder/up_blocks.3/resnets.1/conv1/Conv_output_0.out2_3_31"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"weight_shape": {
"type": "int",
"value": [
"128",
"3",
"3",
"128"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_26",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_31_out-/decoder/up_blocks.3/resnets.1/conv1/Conv_output_0.out2_3_31"
],
"const_args": [
"GroupNorm_26_wts_10_0_25"
],
"out_args": [
"GroupNorm_26.out14_26.out10_0_25"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"wts_shape": {
"type": "int",
"value": [
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"128"
]
},
"beta_shape": {
"type": "int",
"value": [
"128"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_32-/decoder/up_blocks.3/resnets.1/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_26.out14_26.out10_0_25"
],
"const_args": [
"NhwcConv_32_weight_NHWC"
],
"out_args": [
"NhwcConv_32_out-/decoder/up_blocks.3/resnets.1/conv2/Conv_output_0.out2_3_32"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"weight_shape": {
"type": "int",
"value": [
"128",
"3",
"3",
"128"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.3/resnets.1/Add",
"type": "SDAdd",
"in_args": [
"/decoder/up_blocks.3/resnets.0/Add.out_7_1_12",
"NhwcConv_32_out-/decoder/up_blocks.3/resnets.1/conv2/Conv_output_0.out2_3_32"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.3/resnets.1/Add.out_7_1_13"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_27",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.3/resnets.1/Add.out_7_1_13"
],
"const_args": [
"GroupNorm_27_wts_10_0_26"
],
"out_args": [
"GroupNorm_27.out14_27.out10_0_26"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"wts_shape": {
"type": "int",
"value": [
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"128"
]
},
"beta_shape": {
"type": "int",
"value": [
"128"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_33-/decoder/up_blocks.3/resnets.2/conv1/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_27.out14_27.out10_0_26"
],
"const_args": [
"NhwcConv_33_weight_NHWC"
],
"out_args": [
"NhwcConv_33_out-/decoder/up_blocks.3/resnets.2/conv1/Conv_output_0.out2_3_33"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"weight_shape": {
"type": "int",
"value": [
"128",
"3",
"3",
"128"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_28",
"type": "SDGroupNorm",
"in_args": [
"NhwcConv_33_out-/decoder/up_blocks.3/resnets.2/conv1/Conv_output_0.out2_3_33"
],
"const_args": [
"GroupNorm_28_wts_10_0_27"
],
"out_args": [
"GroupNorm_28.out14_28.out10_0_27"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"wts_shape": {
"type": "int",
"value": [
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"128"
]
},
"beta_shape": {
"type": "int",
"value": [
"128"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_34-/decoder/up_blocks.3/resnets.2/conv2/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_28.out14_28.out10_0_27"
],
"const_args": [
"NhwcConv_34_weight_NHWC"
],
"out_args": [
"NhwcConv_34_out-/decoder/up_blocks.3/resnets.2/conv2/Conv_output_0.out2_3_34"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"weight_shape": {
"type": "int",
"value": [
"128",
"3",
"3",
"128"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "/decoder/up_blocks.3/resnets.2/Add",
"type": "SDAdd",
"in_args": [
"/decoder/up_blocks.3/resnets.1/Add.out_7_1_13",
"NhwcConv_34_out-/decoder/up_blocks.3/resnets.2/conv2/Conv_output_0.out2_3_34"
],
"const_args": [],
"out_args": [
"/decoder/up_blocks.3/resnets.2/Add.out_7_1_14"
],
"attrs": {
"a_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"b_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"c_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"is_bias_add": {
"type": "int",
"value": [
"0"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "GroupNorm_29",
"type": "SDGroupNorm",
"in_args": [
"/decoder/up_blocks.3/resnets.2/Add.out_7_1_14"
],
"const_args": [
"GroupNorm_29_wts_10_0_28"
],
"out_args": [
"GroupNorm_29.out14_29.out10_0_28"
],
"attrs": {
"epsilon": {
"type": "float",
"value": [
"9.999999974752427e-07"
]
},
"groups": {
"type": "int",
"value": [
"32"
]
},
"channels_last": {
"type": "int",
"value": [
"1"
]
},
"activation": {
"type": "int",
"value": [
"0"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"wts_shape": {
"type": "int",
"value": [
"256"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfloat16"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"gamma_shape": {
"type": "int",
"value": [
"128"
]
},
"beta_shape": {
"type": "int",
"value": [
"128"
]
},
"nonlinear": {
"type": "str",
"value": [
"Silu"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
},
{
"name": "NhwcConv_35-/decoder/conv_out/Conv",
"type": "SDConv",
"in_args": [
"GroupNorm_29.out14_29.out10_0_28"
],
"const_args": [
"NhwcConv_35_weight_NHWC"
],
"out_args": [
"NhwcConv_35_out-images.out2_3_35"
],
"attrs": {
"auto_pad": {
"type": "str",
"value": [
"NOTSET"
]
},
"dilations": {
"type": "int",
"value": [
"1",
"1"
]
},
"group": {
"type": "int",
"value": [
"1"
]
},
"kernel_shape": {
"type": "int",
"value": [
"3",
"3"
]
},
"pads": {
"type": "int",
"value": [
"1",
"1",
"1",
"1"
]
},
"strides": {
"type": "int",
"value": [
"1",
"1"
]
},
"input_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"128"
]
},
"output_shape": {
"type": "str",
"value": [
"1",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"3"
]
},
"weight_shape": {
"type": "int",
"value": [
"3",
"3",
"3",
"128"
]
},
"out_dtypes": {
"type": "str",
"value": [
"bfloat16"
]
},
"in_dtypes": {
"type": "str",
"value": [
"bfloat16",
"bfp16ebs8",
"float"
]
},
"ctrl_packet": {
"type": "int",
"value": [
"1"
]
}
}
}
],
"fused_tensors": {
"in": {
"buffer_size": 64,
"xrt_arg_id": 0,
"packed_tensors": [
"Transpose_0_out-latents.out2_3_0"
]
},
"out": {
"buffer_size": 8,
"xrt_arg_id": 1,
"packed_tensors": [
"NhwcConv_35_out-images.out2_3_35"
]
},
"scratch": {
"buffer_size": 67648,
"xrt_arg_id": 2,
"packed_tensors": [
"NhwcConv_0_out-/post_quant_conv/Conv_output_0.out2_3_0",
"NhwcConv_1_out-/decoder/conv_in/Conv_output_0.out2_3_1",
"GroupNorm_0.out14_0.out10_0_0",
"NhwcConv_2_out-/decoder/mid_block/resnets.0/conv1/Conv_output_0.out2_3_2",
"GroupNorm_1.out14_1.out10_0_1",
"NhwcConv_3_out-/decoder/mid_block/resnets.0/conv2/Conv_output_0.out2_3_3",
"/decoder/mid_block/resnets.0/Add.out_7_1_0",
"/decoder/mid_block/attentions.0/group_norm/Add_output_0_4d_NHWC.out10_1_0",
"/decoder/mid_block/attentions.0/to_k/Add_output_0.out3_5_0",
"/decoder/mid_block/attentions.0/to_q/Add_output_0.out3_5_1",
"/decoder/mid_block/attentions.0/to_v/Add_output_0.out3_5_2",
"/decoder/mid_block/attentions.0/Reshape_4_output_0.out9_0",
"/decoder/mid_block/attentions.0/Transpose_6_output_0_4d.out3_1_0",
"/decoder/mid_block/attentions.0/Add_output_0_NHWC.out_7_1_1",
"GroupNorm_3.out14_3.out10_0_2",
"NhwcConv_4_out-/decoder/mid_block/resnets.1/conv1/Conv_output_0.out2_3_4",
"GroupNorm_4.out14_4.out10_0_3",
"NhwcConv_5_out-/decoder/mid_block/resnets.1/conv2/Conv_output_0.out2_3_5",
"/decoder/mid_block/resnets.1/Add_output_0_NHWC.out_7_1_2",
"GroupNorm_5.out14_5.out10_0_4",
"NhwcConv_6_out-/decoder/up_blocks.0/resnets.0/conv1/Conv_output_0.out2_3_6",
"GroupNorm_6.out14_6.out10_0_5",
"NhwcConv_7_out-/decoder/up_blocks.0/resnets.0/conv2/Conv_output_0.out2_3_7",
"/decoder/up_blocks.0/resnets.0/Add_output_0_NHWC.out_7_1_3",
"GroupNorm_7.out14_7.out10_0_6",
"NhwcConv_8_out-/decoder/up_blocks.0/resnets.1/conv1/Conv_output_0.out2_3_8",
"GroupNorm_8.out14_8.out10_0_7",
"NhwcConv_9_out-/decoder/up_blocks.0/resnets.1/conv2/Conv_output_0.out2_3_9",
"/decoder/up_blocks.0/resnets.1/Add_output_0_NHWC.out_7_1_4",
"GroupNorm_9.out14_9.out10_0_8",
"NhwcConv_10_out-/decoder/up_blocks.0/resnets.2/conv1/Conv_output_0.out2_3_10",
"GroupNorm_10.out14_10.out10_0_9",
"NhwcConv_11_out-/decoder/up_blocks.0/resnets.2/conv2/Conv_output_0.out2_3_11",
"/decoder/up_blocks.0/resnets.2/Add_output_0.nhwc7_0.out_7_1_5",
"/decoder/up_blocks.0/upsamplers.0/Resize_output_0.nhwc7_0.out_6_0",
"NhwcConv_12_out-/decoder/up_blocks.0/upsamplers.0/conv/Conv_output_0.out2_3_12",
"GroupNorm_11.out14_11.out10_0_10",
"NhwcConv_13_out-/decoder/up_blocks.1/resnets.0/conv1/Conv_output_0.out2_3_13",
"GroupNorm_12.out14_12.out10_0_11",
"NhwcConv_14_out-/decoder/up_blocks.1/resnets.0/conv2/Conv_output_0.out2_3_14",
"/decoder/up_blocks.1/resnets.0/Add.out_7_1_6",
"GroupNorm_13.out14_13.out10_0_12",
"NhwcConv_15_out-/decoder/up_blocks.1/resnets.1/conv1/Conv_output_0.out2_3_15",
"GroupNorm_14.out14_14.out10_0_13",
"NhwcConv_16_out-/decoder/up_blocks.1/resnets.1/conv2/Conv_output_0.out2_3_16",
"/decoder/up_blocks.1/resnets.1/Add.out_7_1_7",
"GroupNorm_15.out14_15.out10_0_14",
"NhwcConv_17_out-/decoder/up_blocks.1/resnets.2/conv1/Conv_output_0.out2_3_17",
"GroupNorm_16.out14_16.out10_0_15",
"NhwcConv_18_out-/decoder/up_blocks.1/resnets.2/conv2/Conv_output_0.out2_3_18",
"/decoder/up_blocks.1/resnets.2/Add.out_7_1_8",
"/decoder/up_blocks.1/upsamplers.0/Resize_output_0.nhwc7_1.out_6_1",
"NhwcConv_19_out-/decoder/up_blocks.1/upsamplers.0/conv/Conv_output_0.out2_3_19",
"GroupNorm_17.out14_17.out10_0_16",
"NhwcConv_20_out-/decoder/up_blocks.2/resnets.0/conv_shortcut/Conv_output_0.out2_3_20",
"NhwcConv_21_out-/decoder/up_blocks.2/resnets.0/conv1/Conv_output_0.out2_3_21",
"GroupNorm_18.out14_18.out10_0_17",
"NhwcConv_22_out-/decoder/up_blocks.2/resnets.0/conv2/Conv_output_0.out2_3_22",
"/decoder/up_blocks.2/resnets.0/Add.out_7_1_9",
"GroupNorm_19.out14_19.out10_0_18",
"NhwcConv_23_out-/decoder/up_blocks.2/resnets.1/conv1/Conv_output_0.out2_3_23",
"GroupNorm_20.out14_20.out10_0_19",
"NhwcConv_24_out-/decoder/up_blocks.2/resnets.1/conv2/Conv_output_0.out2_3_24",
"/decoder/up_blocks.2/resnets.1/Add.out_7_1_10",
"GroupNorm_21.out14_21.out10_0_20",
"NhwcConv_25_out-/decoder/up_blocks.2/resnets.2/conv1/Conv_output_0.out2_3_25",
"GroupNorm_22.out14_22.out10_0_21",
"NhwcConv_26_out-/decoder/up_blocks.2/resnets.2/conv2/Conv_output_0.out2_3_26",
"/decoder/up_blocks.2/resnets.2/Add.out_7_1_11",
"/decoder/up_blocks.2/upsamplers.0/Resize_output_0.nhwc7_2.out_6_2",
"NhwcConv_27_out-/decoder/up_blocks.2/upsamplers.0/conv/Conv_output_0.out2_3_27",
"GroupNorm_23.out14_23.out10_0_22",
"NhwcConv_28_out-/decoder/up_blocks.3/resnets.0/conv_shortcut/Conv_output_0.out2_3_28",
"NhwcConv_29_out-/decoder/up_blocks.3/resnets.0/conv1/Conv_output_0.out2_3_29",
"GroupNorm_24.out14_24.out10_0_23",
"NhwcConv_30_out-/decoder/up_blocks.3/resnets.0/conv2/Conv_output_0.out2_3_30",
"/decoder/up_blocks.3/resnets.0/Add.out_7_1_12",
"GroupNorm_25.out14_25.out10_0_24",
"NhwcConv_31_out-/decoder/up_blocks.3/resnets.1/conv1/Conv_output_0.out2_3_31",
"GroupNorm_26.out14_26.out10_0_25",
"NhwcConv_32_out-/decoder/up_blocks.3/resnets.1/conv2/Conv_output_0.out2_3_32",
"/decoder/up_blocks.3/resnets.1/Add.out_7_1_13",
"GroupNorm_27.out14_27.out10_0_26",
"NhwcConv_33_out-/decoder/up_blocks.3/resnets.2/conv1/Conv_output_0.out2_3_33",
"GroupNorm_28.out14_28.out10_0_27",
"NhwcConv_34_out-/decoder/up_blocks.3/resnets.2/conv2/Conv_output_0.out2_3_34",
"/decoder/up_blocks.3/resnets.2/Add.out_7_1_14",
"GroupNorm_29.out14_29.out10_0_28"
]
},
"const": {
"buffer_size": 61249280,
"xrt_arg_id": 3,
"packed_tensors": [
"NhwcConv_0_weight_NHWC",
"NhwcConv_1_weight_NHWC",
"GroupNorm_0_wts_10_0_0",
"NhwcConv_2_weight_NHWC",
"GroupNorm_1_wts_10_0_1",
"NhwcConv_3_weight_NHWC",
"GroupNorm_2_wts_10_1_0",
"onnx::MatMul_920_15_0_0",
"onnx::MatMul_919_15_0_1",
"onnx::MatMul_921_15_0_2",
"/decoder/mid_block/attentions.0/MatMulmha_5_0_0_mask.9_0",
"onnx::MatMul_929_15_0_3",
"GroupNorm_3_wts_10_0_2",
"NhwcConv_4_weight_NHWC",
"GroupNorm_4_wts_10_0_3",
"NhwcConv_5_weight_NHWC",
"GroupNorm_5_wts_10_0_4",
"NhwcConv_6_weight_NHWC",
"GroupNorm_6_wts_10_0_5",
"NhwcConv_7_weight_NHWC",
"GroupNorm_7_wts_10_0_6",
"NhwcConv_8_weight_NHWC",
"GroupNorm_8_wts_10_0_7",
"NhwcConv_9_weight_NHWC",
"GroupNorm_9_wts_10_0_8",
"NhwcConv_10_weight_NHWC",
"GroupNorm_10_wts_10_0_9",
"NhwcConv_11_weight_NHWC",
"/decoder/up_blocks.0/upsamplers.0/Resize.weights6_0",
"NhwcConv_12_weight_NHWC",
"GroupNorm_11_wts_10_0_10",
"NhwcConv_13_weight_NHWC",
"GroupNorm_12_wts_10_0_11",
"NhwcConv_14_weight_NHWC",
"GroupNorm_13_wts_10_0_12",
"NhwcConv_15_weight_NHWC",
"GroupNorm_14_wts_10_0_13",
"NhwcConv_16_weight_NHWC",
"GroupNorm_15_wts_10_0_14",
"NhwcConv_17_weight_NHWC",
"GroupNorm_16_wts_10_0_15",
"NhwcConv_18_weight_NHWC",
"/decoder/up_blocks.1/upsamplers.0/Resize.weights6_1",
"NhwcConv_19_weight_NHWC",
"GroupNorm_17_wts_10_0_16",
"NhwcConv_20_weight_NHWC",
"NhwcConv_21_weight_NHWC",
"GroupNorm_18_wts_10_0_17",
"NhwcConv_22_weight_NHWC",
"GroupNorm_19_wts_10_0_18",
"NhwcConv_23_weight_NHWC",
"GroupNorm_20_wts_10_0_19",
"NhwcConv_24_weight_NHWC",
"GroupNorm_21_wts_10_0_20",
"NhwcConv_25_weight_NHWC",
"GroupNorm_22_wts_10_0_21",
"NhwcConv_26_weight_NHWC",
"/decoder/up_blocks.2/upsamplers.0/Resize.weights6_2",
"NhwcConv_27_weight_NHWC",
"GroupNorm_23_wts_10_0_22",
"NhwcConv_28_weight_NHWC",
"NhwcConv_29_weight_NHWC",
"GroupNorm_24_wts_10_0_23",
"NhwcConv_30_weight_NHWC",
"GroupNorm_25_wts_10_0_24",
"NhwcConv_31_weight_NHWC",
"GroupNorm_26_wts_10_0_25",
"NhwcConv_32_weight_NHWC",
"GroupNorm_27_wts_10_0_26",
"NhwcConv_33_weight_NHWC",
"GroupNorm_28_wts_10_0_27",
"NhwcConv_34_weight_NHWC",
"GroupNorm_29_wts_10_0_28",
"NhwcConv_35_weight_NHWC"
]
},
"super_instr": {
"buffer_size": 0,
"xrt_arg_id": 4,
"packed_tensors": []
}
},
"tensor_map": {
"Transpose_0_out-latents.out2_3_0": {
"packed_buffer_label": "in",
"xrt_arg_id": 0,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
32
],
"size_in_bytes": 64,
"op_tensor_size": 64,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 0
},
"NhwcConv_35_out-images.out2_3_35": {
"packed_buffer_label": "out",
"xrt_arg_id": 1,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
3
],
"size_in_bytes": 6,
"op_tensor_size": 6,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 0
},
"NhwcConv_0_out-/post_quant_conv/Conv_output_0.out2_3_0": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
32
],
"size_in_bytes": 64,
"op_tensor_size": 64,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 0
},
"NhwcConv_1_out-/decoder/conv_in/Conv_output_0.out2_3_1": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 64
},
"GroupNorm_0.out14_0.out10_0_0": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 1088
},
"NhwcConv_2_out-/decoder/mid_block/resnets.0/conv1/Conv_output_0.out2_3_2": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 2112
},
"GroupNorm_1.out14_1.out10_0_1": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 3136
},
"NhwcConv_3_out-/decoder/mid_block/resnets.0/conv2/Conv_output_0.out2_3_3": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 4160
},
"/decoder/mid_block/resnets.0/Add.out_7_1_0": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 5184
},
"/decoder/mid_block/attentions.0/group_norm/Add_output_0_4d_NHWC.out10_1_0": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h*latent_w",
"False"
],
"offset": 6208
},
"/decoder/mid_block/attentions.0/to_k/Add_output_0.out3_5_0": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h*latent_w",
"False"
],
"offset": 7232
},
"/decoder/mid_block/attentions.0/to_q/Add_output_0.out3_5_1": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h*latent_w",
"False"
],
"offset": 8256
},
"/decoder/mid_block/attentions.0/to_v/Add_output_0.out3_5_2": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h*latent_w",
"False"
],
"offset": 9280
},
"/decoder/mid_block/attentions.0/Reshape_4_output_0.out9_0": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h*latent_w",
"False"
],
"offset": 10304
},
"/decoder/mid_block/attentions.0/Transpose_6_output_0_4d.out3_1_0": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 11328
},
"/decoder/mid_block/attentions.0/Add_output_0_NHWC.out_7_1_1": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 12352
},
"GroupNorm_3.out14_3.out10_0_2": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 13376
},
"NhwcConv_4_out-/decoder/mid_block/resnets.1/conv1/Conv_output_0.out2_3_4": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 14400
},
"GroupNorm_4.out14_4.out10_0_3": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 15424
},
"NhwcConv_5_out-/decoder/mid_block/resnets.1/conv2/Conv_output_0.out2_3_5": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 16448
},
"/decoder/mid_block/resnets.1/Add_output_0_NHWC.out_7_1_2": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 17472
},
"GroupNorm_5.out14_5.out10_0_4": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 18496
},
"NhwcConv_6_out-/decoder/up_blocks.0/resnets.0/conv1/Conv_output_0.out2_3_6": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 19520
},
"GroupNorm_6.out14_6.out10_0_5": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 20544
},
"NhwcConv_7_out-/decoder/up_blocks.0/resnets.0/conv2/Conv_output_0.out2_3_7": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 21568
},
"/decoder/up_blocks.0/resnets.0/Add_output_0_NHWC.out_7_1_3": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 22592
},
"GroupNorm_7.out14_7.out10_0_6": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 23616
},
"NhwcConv_8_out-/decoder/up_blocks.0/resnets.1/conv1/Conv_output_0.out2_3_8": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 24640
},
"GroupNorm_8.out14_8.out10_0_7": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 25664
},
"NhwcConv_9_out-/decoder/up_blocks.0/resnets.1/conv2/Conv_output_0.out2_3_9": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 26688
},
"/decoder/up_blocks.0/resnets.1/Add_output_0_NHWC.out_7_1_4": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 27712
},
"GroupNorm_9.out14_9.out10_0_8": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 28736
},
"NhwcConv_10_out-/decoder/up_blocks.0/resnets.2/conv1/Conv_output_0.out2_3_10": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 29760
},
"GroupNorm_10.out14_10.out10_0_9": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 30784
},
"NhwcConv_11_out-/decoder/up_blocks.0/resnets.2/conv2/Conv_output_0.out2_3_11": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 31808
},
"/decoder/up_blocks.0/resnets.2/Add_output_0.nhwc7_0.out_7_1_5": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"latent_h",
"latent_w",
"False"
],
"offset": 32832
},
"/decoder/up_blocks.0/upsamplers.0/Resize_output_0.nhwc7_0.out_6_0": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 33856
},
"NhwcConv_12_out-/decoder/up_blocks.0/upsamplers.0/conv/Conv_output_0.out2_3_12": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 34880
},
"GroupNorm_11.out14_11.out10_0_10": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 35904
},
"NhwcConv_13_out-/decoder/up_blocks.1/resnets.0/conv1/Conv_output_0.out2_3_13": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 36928
},
"GroupNorm_12.out14_12.out10_0_11": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 37952
},
"NhwcConv_14_out-/decoder/up_blocks.1/resnets.0/conv2/Conv_output_0.out2_3_14": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 38976
},
"/decoder/up_blocks.1/resnets.0/Add.out_7_1_6": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 40000
},
"GroupNorm_13.out14_13.out10_0_12": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 41024
},
"NhwcConv_15_out-/decoder/up_blocks.1/resnets.1/conv1/Conv_output_0.out2_3_15": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 42048
},
"GroupNorm_14.out14_14.out10_0_13": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 43072
},
"NhwcConv_16_out-/decoder/up_blocks.1/resnets.1/conv2/Conv_output_0.out2_3_16": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 44096
},
"/decoder/up_blocks.1/resnets.1/Add.out_7_1_7": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 45120
},
"GroupNorm_15.out14_15.out10_0_14": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 46144
},
"NhwcConv_17_out-/decoder/up_blocks.1/resnets.2/conv1/Conv_output_0.out2_3_17": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 47168
},
"GroupNorm_16.out14_16.out10_0_15": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 48192
},
"NhwcConv_18_out-/decoder/up_blocks.1/resnets.2/conv2/Conv_output_0.out2_3_18": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 49216
},
"/decoder/up_blocks.1/resnets.2/Add.out_7_1_8": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*latent_h)",
"floor(2.0*latent_w)",
"False"
],
"offset": 50240
},
"/decoder/up_blocks.1/upsamplers.0/Resize_output_0.nhwc7_1.out_6_1": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 51264
},
"NhwcConv_19_out-/decoder/up_blocks.1/upsamplers.0/conv/Conv_output_0.out2_3_19": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 52288
},
"GroupNorm_17.out14_17.out10_0_16": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 53312
},
"NhwcConv_20_out-/decoder/up_blocks.2/resnets.0/conv_shortcut/Conv_output_0.out2_3_20": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 54336
},
"NhwcConv_21_out-/decoder/up_blocks.2/resnets.0/conv1/Conv_output_0.out2_3_21": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 54848
},
"GroupNorm_18.out14_18.out10_0_17": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 55360
},
"NhwcConv_22_out-/decoder/up_blocks.2/resnets.0/conv2/Conv_output_0.out2_3_22": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 55872
},
"/decoder/up_blocks.2/resnets.0/Add.out_7_1_9": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 56384
},
"GroupNorm_19.out14_19.out10_0_18": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 56896
},
"NhwcConv_23_out-/decoder/up_blocks.2/resnets.1/conv1/Conv_output_0.out2_3_23": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 57408
},
"GroupNorm_20.out14_20.out10_0_19": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 57920
},
"NhwcConv_24_out-/decoder/up_blocks.2/resnets.1/conv2/Conv_output_0.out2_3_24": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 58432
},
"/decoder/up_blocks.2/resnets.1/Add.out_7_1_10": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 58944
},
"GroupNorm_21.out14_21.out10_0_20": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 59456
},
"NhwcConv_25_out-/decoder/up_blocks.2/resnets.2/conv1/Conv_output_0.out2_3_25": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 59968
},
"GroupNorm_22.out14_22.out10_0_21": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 60480
},
"NhwcConv_26_out-/decoder/up_blocks.2/resnets.2/conv2/Conv_output_0.out2_3_26": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 60992
},
"/decoder/up_blocks.2/resnets.2/Add.out_7_1_11": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*latent_h))",
"floor(2.0*floor(2.0*latent_w))",
"False"
],
"offset": 61504
},
"/decoder/up_blocks.2/upsamplers.0/Resize_output_0.nhwc7_2.out_6_2": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 62016
},
"NhwcConv_27_out-/decoder/up_blocks.2/upsamplers.0/conv/Conv_output_0.out2_3_27": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 62528
},
"GroupNorm_23.out14_23.out10_0_22": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 63040
},
"NhwcConv_28_out-/decoder/up_blocks.3/resnets.0/conv_shortcut/Conv_output_0.out2_3_28": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 63552
},
"NhwcConv_29_out-/decoder/up_blocks.3/resnets.0/conv1/Conv_output_0.out2_3_29": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 63808
},
"GroupNorm_24.out14_24.out10_0_23": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 64064
},
"NhwcConv_30_out-/decoder/up_blocks.3/resnets.0/conv2/Conv_output_0.out2_3_30": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 64320
},
"/decoder/up_blocks.3/resnets.0/Add.out_7_1_12": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 64576
},
"GroupNorm_25.out14_25.out10_0_24": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 64832
},
"NhwcConv_31_out-/decoder/up_blocks.3/resnets.1/conv1/Conv_output_0.out2_3_31": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 65088
},
"GroupNorm_26.out14_26.out10_0_25": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 65344
},
"NhwcConv_32_out-/decoder/up_blocks.3/resnets.1/conv2/Conv_output_0.out2_3_32": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 65600
},
"/decoder/up_blocks.3/resnets.1/Add.out_7_1_13": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 65856
},
"GroupNorm_27.out14_27.out10_0_26": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 66112
},
"NhwcConv_33_out-/decoder/up_blocks.3/resnets.2/conv1/Conv_output_0.out2_3_33": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 66368
},
"GroupNorm_28.out14_28.out10_0_27": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 66624
},
"NhwcConv_34_out-/decoder/up_blocks.3/resnets.2/conv2/Conv_output_0.out2_3_34": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 66880
},
"/decoder/up_blocks.3/resnets.2/Add.out_7_1_14": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 67136
},
"GroupNorm_29.out14_29.out10_0_28": {
"packed_buffer_label": "scratch",
"xrt_arg_id": 2,
"dtype": "bfloat16",
"shape": [
1,
1,
1,
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"dynamic_shapes": [
"False",
"floor(2.0*floor(2.0*floor(2.0*latent_h)))",
"floor(2.0*floor(2.0*floor(2.0*latent_w)))",
"False"
],
"offset": 67392
},
"NhwcConv_0_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2816
],
"size_in_bytes": 2816,
"op_tensor_size": 2816,
"offset": 0,
"file_name": "cache/NhwcConv_0-post_quant_convConv_0.const",
"file_size": 2816
},
"NhwcConv_1_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
182272
],
"size_in_bytes": 182272,
"op_tensor_size": 182272,
"offset": 2816,
"file_name": "cache/NhwcConv_0-post_quant_convConv_1.const",
"file_size": 182272
},
"GroupNorm_0_wts_10_0_0": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 185088,
"file_name": "cache/NhwcConv_0-post_quant_convConv_2.const",
"file_size": 2048
},
"NhwcConv_2_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 187136,
"file_name": "cache/NhwcConv_0-post_quant_convConv_3.const",
"file_size": 2916352
},
"GroupNorm_1_wts_10_0_1": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 3103488,
"file_name": "cache/NhwcConv_0-post_quant_convConv_4.const",
"file_size": 2048
},
"NhwcConv_3_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 3105536,
"file_name": "cache/NhwcConv_0-post_quant_convConv_5.const",
"file_size": 2916352
},
"GroupNorm_2_wts_10_1_0": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 6021888,
"file_name": "cache/NhwcConv_0-post_quant_convConv_6.const",
"file_size": 2048
},
"onnx::MatMul_920_15_0_0": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
299008
],
"size_in_bytes": 299008,
"op_tensor_size": 299008,
"offset": 6023936,
"file_name": "cache/NhwcConv_0-post_quant_convConv_7.const",
"file_size": 299008
},
"onnx::MatMul_919_15_0_1": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
299008
],
"size_in_bytes": 299008,
"op_tensor_size": 299008,
"offset": 6322944,
"file_name": "cache/NhwcConv_0-post_quant_convConv_8.const",
"file_size": 299008
},
"onnx::MatMul_921_15_0_2": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
299008
],
"size_in_bytes": 299008,
"op_tensor_size": 299008,
"offset": 6621952,
"file_name": "cache/NhwcConv_0-post_quant_convConv_9.const",
"file_size": 299008
},
"/decoder/mid_block/attentions.0/MatMulmha_5_0_0_mask.9_0": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
16384
],
"size_in_bytes": 32768,
"op_tensor_size": 32768,
"offset": 6920960,
"file_name": "cache/NhwcConv_0-post_quant_convConv_10.const",
"file_size": 32768
},
"onnx::MatMul_929_15_0_3": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
299008
],
"size_in_bytes": 299008,
"op_tensor_size": 299008,
"offset": 6953728,
"file_name": "cache/NhwcConv_0-post_quant_convConv_11.const",
"file_size": 299008
},
"GroupNorm_3_wts_10_0_2": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 7252736,
"file_name": "cache/NhwcConv_0-post_quant_convConv_12.const",
"file_size": 2048
},
"NhwcConv_4_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 7254784,
"file_name": "cache/NhwcConv_0-post_quant_convConv_13.const",
"file_size": 2916352
},
"GroupNorm_4_wts_10_0_3": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 10171136,
"file_name": "cache/NhwcConv_0-post_quant_convConv_14.const",
"file_size": 2048
},
"NhwcConv_5_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 10173184,
"file_name": "cache/NhwcConv_0-post_quant_convConv_15.const",
"file_size": 2916352
},
"GroupNorm_5_wts_10_0_4": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 13089536,
"file_name": "cache/NhwcConv_0-post_quant_convConv_16.const",
"file_size": 2048
},
"NhwcConv_6_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 13091584,
"file_name": "cache/NhwcConv_0-post_quant_convConv_17.const",
"file_size": 2916352
},
"GroupNorm_6_wts_10_0_5": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 16007936,
"file_name": "cache/NhwcConv_0-post_quant_convConv_18.const",
"file_size": 2048
},
"NhwcConv_7_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 16009984,
"file_name": "cache/NhwcConv_0-post_quant_convConv_19.const",
"file_size": 2916352
},
"GroupNorm_7_wts_10_0_6": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 18926336,
"file_name": "cache/NhwcConv_0-post_quant_convConv_20.const",
"file_size": 2048
},
"NhwcConv_8_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 18928384,
"file_name": "cache/NhwcConv_0-post_quant_convConv_21.const",
"file_size": 2916352
},
"GroupNorm_8_wts_10_0_7": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 21844736,
"file_name": "cache/NhwcConv_0-post_quant_convConv_22.const",
"file_size": 2048
},
"NhwcConv_9_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 21846784,
"file_name": "cache/NhwcConv_0-post_quant_convConv_23.const",
"file_size": 2916352
},
"GroupNorm_9_wts_10_0_8": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 24763136,
"file_name": "cache/NhwcConv_0-post_quant_convConv_24.const",
"file_size": 2048
},
"NhwcConv_10_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 24765184,
"file_name": "cache/NhwcConv_0-post_quant_convConv_25.const",
"file_size": 2916352
},
"GroupNorm_10_wts_10_0_9": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 27681536,
"file_name": "cache/NhwcConv_0-post_quant_convConv_26.const",
"file_size": 2048
},
"NhwcConv_11_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 27683584,
"file_name": "cache/NhwcConv_0-post_quant_convConv_27.const",
"file_size": 2916352
},
"/decoder/up_blocks.0/upsamplers.0/Resize.weights6_0": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"offset": 30599936,
"file_name": "cache/NhwcConv_0-post_quant_convConv_28.const",
"file_size": 256
},
"NhwcConv_12_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 30600192,
"file_name": "cache/NhwcConv_0-post_quant_convConv_29.const",
"file_size": 2916352
},
"GroupNorm_11_wts_10_0_10": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 33516544,
"file_name": "cache/NhwcConv_0-post_quant_convConv_30.const",
"file_size": 2048
},
"NhwcConv_13_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 33518592,
"file_name": "cache/NhwcConv_0-post_quant_convConv_31.const",
"file_size": 2916352
},
"GroupNorm_12_wts_10_0_11": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 36434944,
"file_name": "cache/NhwcConv_0-post_quant_convConv_32.const",
"file_size": 2048
},
"NhwcConv_14_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 36436992,
"file_name": "cache/NhwcConv_0-post_quant_convConv_33.const",
"file_size": 2916352
},
"GroupNorm_13_wts_10_0_12": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 39353344,
"file_name": "cache/NhwcConv_0-post_quant_convConv_34.const",
"file_size": 2048
},
"NhwcConv_15_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 39355392,
"file_name": "cache/NhwcConv_0-post_quant_convConv_35.const",
"file_size": 2916352
},
"GroupNorm_14_wts_10_0_13": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 42271744,
"file_name": "cache/NhwcConv_0-post_quant_convConv_36.const",
"file_size": 2048
},
"NhwcConv_16_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 42273792,
"file_name": "cache/NhwcConv_0-post_quant_convConv_37.const",
"file_size": 2916352
},
"GroupNorm_15_wts_10_0_14": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 45190144,
"file_name": "cache/NhwcConv_0-post_quant_convConv_38.const",
"file_size": 2048
},
"NhwcConv_17_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 45192192,
"file_name": "cache/NhwcConv_0-post_quant_convConv_39.const",
"file_size": 2916352
},
"GroupNorm_16_wts_10_0_15": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 48108544,
"file_name": "cache/NhwcConv_0-post_quant_convConv_40.const",
"file_size": 2048
},
"NhwcConv_18_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 48110592,
"file_name": "cache/NhwcConv_0-post_quant_convConv_41.const",
"file_size": 2916352
},
"/decoder/up_blocks.1/upsamplers.0/Resize.weights6_1": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"offset": 51026944,
"file_name": "cache/NhwcConv_0-post_quant_convConv_42.const",
"file_size": 256
},
"NhwcConv_19_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
2916352
],
"size_in_bytes": 2916352,
"op_tensor_size": 2916352,
"offset": 51027200,
"file_name": "cache/NhwcConv_0-post_quant_convConv_43.const",
"file_size": 2916352
},
"GroupNorm_17_wts_10_0_16": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
1024
],
"size_in_bytes": 2048,
"op_tensor_size": 2048,
"offset": 53943552,
"file_name": "cache/NhwcConv_0-post_quant_convConv_44.const",
"file_size": 2048
},
"NhwcConv_20_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
163840
],
"size_in_bytes": 163840,
"op_tensor_size": 163840,
"offset": 53945600,
"file_name": "cache/NhwcConv_0-post_quant_convConv_45.const",
"file_size": 163840
},
"NhwcConv_21_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
1458176
],
"size_in_bytes": 1458176,
"op_tensor_size": 1458176,
"offset": 54109440,
"file_name": "cache/NhwcConv_0-post_quant_convConv_46.const",
"file_size": 1458176
},
"GroupNorm_18_wts_10_0_17": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"offset": 55567616,
"file_name": "cache/NhwcConv_0-post_quant_convConv_47.const",
"file_size": 1024
},
"NhwcConv_22_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
729088
],
"size_in_bytes": 729088,
"op_tensor_size": 729088,
"offset": 55568640,
"file_name": "cache/NhwcConv_0-post_quant_convConv_48.const",
"file_size": 729088
},
"GroupNorm_19_wts_10_0_18": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"offset": 56297728,
"file_name": "cache/NhwcConv_0-post_quant_convConv_49.const",
"file_size": 1024
},
"NhwcConv_23_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
729088
],
"size_in_bytes": 729088,
"op_tensor_size": 729088,
"offset": 56298752,
"file_name": "cache/NhwcConv_0-post_quant_convConv_50.const",
"file_size": 729088
},
"GroupNorm_20_wts_10_0_19": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"offset": 57027840,
"file_name": "cache/NhwcConv_0-post_quant_convConv_51.const",
"file_size": 1024
},
"NhwcConv_24_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
729088
],
"size_in_bytes": 729088,
"op_tensor_size": 729088,
"offset": 57028864,
"file_name": "cache/NhwcConv_0-post_quant_convConv_52.const",
"file_size": 729088
},
"GroupNorm_21_wts_10_0_20": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"offset": 57757952,
"file_name": "cache/NhwcConv_0-post_quant_convConv_53.const",
"file_size": 1024
},
"NhwcConv_25_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
729088
],
"size_in_bytes": 729088,
"op_tensor_size": 729088,
"offset": 57758976,
"file_name": "cache/NhwcConv_0-post_quant_convConv_54.const",
"file_size": 729088
},
"GroupNorm_22_wts_10_0_21": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"offset": 58488064,
"file_name": "cache/NhwcConv_0-post_quant_convConv_55.const",
"file_size": 1024
},
"NhwcConv_26_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
729088
],
"size_in_bytes": 729088,
"op_tensor_size": 729088,
"offset": 58489088,
"file_name": "cache/NhwcConv_0-post_quant_convConv_56.const",
"file_size": 729088
},
"/decoder/up_blocks.2/upsamplers.0/Resize.weights6_2": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
128
],
"size_in_bytes": 256,
"op_tensor_size": 256,
"offset": 59218176,
"file_name": "cache/NhwcConv_0-post_quant_convConv_57.const",
"file_size": 256
},
"NhwcConv_27_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
729088
],
"size_in_bytes": 729088,
"op_tensor_size": 729088,
"offset": 59218432,
"file_name": "cache/NhwcConv_0-post_quant_convConv_58.const",
"file_size": 729088
},
"GroupNorm_23_wts_10_0_22": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
512
],
"size_in_bytes": 1024,
"op_tensor_size": 1024,
"offset": 59947520,
"file_name": "cache/NhwcConv_0-post_quant_convConv_59.const",
"file_size": 1024
},
"NhwcConv_28_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
40960
],
"size_in_bytes": 40960,
"op_tensor_size": 40960,
"offset": 59948544,
"file_name": "cache/NhwcConv_0-post_quant_convConv_60.const",
"file_size": 40960
},
"NhwcConv_29_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
364544
],
"size_in_bytes": 364544,
"op_tensor_size": 364544,
"offset": 59989504,
"file_name": "cache/NhwcConv_0-post_quant_convConv_61.const",
"file_size": 364544
},
"GroupNorm_24_wts_10_0_23": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"offset": 60354048,
"file_name": "cache/NhwcConv_0-post_quant_convConv_62.const",
"file_size": 512
},
"NhwcConv_30_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
174080
],
"size_in_bytes": 174080,
"op_tensor_size": 174080,
"offset": 60354560,
"file_name": "cache/NhwcConv_0-post_quant_convConv_63.const",
"file_size": 174080
},
"GroupNorm_25_wts_10_0_24": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"offset": 60528640,
"file_name": "cache/NhwcConv_0-post_quant_convConv_64.const",
"file_size": 512
},
"NhwcConv_31_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
174080
],
"size_in_bytes": 174080,
"op_tensor_size": 174080,
"offset": 60529152,
"file_name": "cache/NhwcConv_0-post_quant_convConv_65.const",
"file_size": 174080
},
"GroupNorm_26_wts_10_0_25": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"offset": 60703232,
"file_name": "cache/NhwcConv_0-post_quant_convConv_66.const",
"file_size": 512
},
"NhwcConv_32_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
174080
],
"size_in_bytes": 174080,
"op_tensor_size": 174080,
"offset": 60703744,
"file_name": "cache/NhwcConv_0-post_quant_convConv_67.const",
"file_size": 174080
},
"GroupNorm_27_wts_10_0_26": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"offset": 60877824,
"file_name": "cache/NhwcConv_0-post_quant_convConv_68.const",
"file_size": 512
},
"NhwcConv_33_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
174080
],
"size_in_bytes": 174080,
"op_tensor_size": 174080,
"offset": 60878336,
"file_name": "cache/NhwcConv_0-post_quant_convConv_69.const",
"file_size": 174080
},
"GroupNorm_28_wts_10_0_27": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"offset": 61052416,
"file_name": "cache/NhwcConv_0-post_quant_convConv_70.const",
"file_size": 512
},
"NhwcConv_34_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
174080
],
"size_in_bytes": 174080,
"op_tensor_size": 174080,
"offset": 61052928,
"file_name": "cache/NhwcConv_0-post_quant_convConv_71.const",
"file_size": 174080
},
"GroupNorm_29_wts_10_0_28": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfloat16",
"shape": [
256
],
"size_in_bytes": 512,
"op_tensor_size": 512,
"offset": 61227008,
"file_name": "cache/NhwcConv_0-post_quant_convConv_72.const",
"file_size": 512
},
"NhwcConv_35_weight_NHWC": {
"packed_buffer_label": "const",
"xrt_arg_id": 3,
"dtype": "bfp16ebs8",
"shape": [
21760
],
"size_in_bytes": 21760,
"op_tensor_size": 21760,
"offset": 61227520,
"file_name": "cache/NhwcConv_0-post_quant_convConv_73.const",
"file_size": 21760
}
},
"dynamic_shape_subgraph": true,
"dynamic_shape_list": [
{
"latent_h": 128,
"floor(2.0*floor(2.0*floor(2.0*latent_w)))": 1024,
"floor(2.0*latent_h)": 256,
"latent_w": 128,
"floor(2.0*latent_w)": 256,
"floor(2.0*floor(2.0*latent_w))": 512,
"floor(2.0*floor(2.0*latent_h))": 512,
"floor(2.0*floor(2.0*floor(2.0*latent_h)))": 1024,
"latent_h*latent_w": 16384
}
],
"aux_info": {}
}