samplefloat32[batch_size,num_channels,height,width]Castgraph_input_cast0Conv/encoder/conv_in/Convfloat16[128,3,3,3]W〈128×3×3×3〉float16[128]B〈128〉TransposeTranspose_NCHW_to_NHWC_0GroupNormGroupNorm_0float32[128]gamma〈128〉float32[128]beta〈128〉TransposeTranspose_NHWC_to_NCHW_0Conv/encoder/down_blocks.0/resnets.0/conv1/Convfloat16[128,128,3,3]W〈128×128×3×3〉float16[128]B〈128〉TransposeTranspose_NCHW_to_NHWC_1GroupNormGroupNorm_1float32[128]gamma〈128〉float32[128]beta〈128〉TransposeTranspose_NHWC_to_NCHW_1Conv/encoder/down_blocks.0/resnets.0/conv2/Convfloat16[128,128,3,3]W〈128×128×3×3〉float16[128]B〈128〉Add/encoder/down_blocks.0/resnets.0/AddTransposeTranspose_NCHW_to_NHWC_2GroupNormGroupNorm_2float32[128]gamma〈128〉float32[128]beta〈128〉TransposeTranspose_NHWC_to_NCHW_2Conv/encoder/down_blocks.0/resnets.1/conv1/Convfloat16[128,128,3,3]W〈128×128×3×3〉float16[128]B〈128〉TransposeTranspose_NCHW_to_NHWC_3GroupNormGroupNorm_3float32[128]gamma〈128〉float32[128]beta〈128〉TransposeTranspose_NHWC_to_NCHW_3Conv/encoder/down_blocks.0/resnets.1/conv2/Convfloat16[128,128,3,3]W〈128×128×3×3〉float16[128]B〈128〉Add/encoder/down_blocks.0/resnets.1/AddPad/encoder/down_blocks.0/downsamplers.0/Padint64[8]pads〈8〉float16constant_value = 0Conv/encoder/down_blocks.0/downsamplers.0/conv/Convfloat16[128,128,3,3]W〈128×128×3×3〉float16[128]B〈128〉Conv/encoder/down_blocks.1/resnets.0/conv_shortcut/Convfloat16[256,128,1,1]W〈256×128×1×1〉float16[256]B〈256〉TransposeTranspose_NCHW_to_NHWC_4GroupNormGroupNorm_4float32[128]gamma〈128〉float32[128]beta〈128〉TransposeTranspose_NHWC_to_NCHW_4Conv/encoder/down_blocks.1/resnets.0/conv1/Convfloat16[256,128,3,3]W〈256×128×3×3〉float16[256]B〈256〉TransposeTranspose_NCHW_to_NHWC_5GroupNormGroupNorm_5float32[256]gamma〈256〉float32[256]beta〈256〉TransposeTranspose_NHWC_to_NCHW_5Conv/encoder/down_blocks.1/resnets.0/conv2/Convfloat16[256,256,3,3]W〈256×256×3×3〉float16[256]B〈256〉Add/encoder/down_blocks.1/resnets.0/AddTransposeTranspose_NCHW_to_NHWC_6GroupNormGroupNorm_6float32[256]gamma〈256〉float32[256]beta〈256〉TransposeTranspose_NHWC_to_NCHW_6Conv/encoder/down_blocks.1/resnets.1/conv1/Convfloat16[256,256,3,3]W〈256×256×3×3〉float16[256]B〈256〉TransposeTranspose_NCHW_to_NHWC_7GroupNormGroupNorm_7float32[256]gamma〈256〉float32[256]beta〈256〉TransposeTranspose_NHWC_to_NCHW_7Conv/encoder/down_blocks.1/resnets.1/conv2/Convfloat16[256,256,3,3]W〈256×256×3×3〉float16[256]B〈256〉Add/encoder/down_blocks.1/resnets.1/AddPad/encoder/down_blocks.1/downsamplers.0/Padint64[8]pads〈8〉float16constant_value = 0Conv/encoder/down_blocks.1/downsamplers.0/conv/Convfloat16[256,256,3,3]W〈256×256×3×3〉float16[256]B〈256〉Conv/encoder/down_blocks.2/resnets.0/conv_shortcut/Convfloat16[512,256,1,1]W〈512×256×1×1〉float16[512]B〈512〉TransposeTranspose_NCHW_to_NHWC_8GroupNormGroupNorm_8float32[256]gamma〈256〉float32[256]beta〈256〉TransposeTranspose_NHWC_to_NCHW_8Conv/encoder/down_blocks.2/resnets.0/conv1/Convfloat16[512,256,3,3]W〈512×256×3×3〉float16[512]B〈512〉TransposeTranspose_NCHW_to_NHWC_9GroupNormGroupNorm_9float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_9Conv/encoder/down_blocks.2/resnets.0/conv2/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉Add/encoder/down_blocks.2/resnets.0/AddTransposeTranspose_NCHW_to_NHWC_10GroupNormGroupNorm_10float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_10Conv/encoder/down_blocks.2/resnets.1/conv1/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉TransposeTranspose_NCHW_to_NHWC_11GroupNormGroupNorm_11float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_11Conv/encoder/down_blocks.2/resnets.1/conv2/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉Add/encoder/down_blocks.2/resnets.1/AddPad/encoder/down_blocks.2/downsamplers.0/Padint64[8]pads〈8〉float16constant_value = 0Conv/encoder/down_blocks.2/downsamplers.0/conv/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉TransposeTranspose_NCHW_to_NHWC_12GroupNormGroupNorm_12float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_12Conv/encoder/down_blocks.3/resnets.0/conv1/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉TransposeTranspose_NCHW_to_NHWC_13GroupNormGroupNorm_13float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_13Conv/encoder/down_blocks.3/resnets.0/conv2/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉Add/encoder/down_blocks.3/resnets.0/AddTransposeTranspose_NCHW_to_NHWC_14GroupNormGroupNorm_14float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_14Conv/encoder/down_blocks.3/resnets.1/conv1/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉TransposeTranspose_NCHW_to_NHWC_15GroupNormGroupNorm_15float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_15Conv/encoder/down_blocks.3/resnets.1/conv2/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉Add/encoder/down_blocks.3/resnets.1/AddTransposeTranspose_NCHW_to_NHWC_16GroupNormGroupNorm_16float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_16Conv/encoder/mid_block/resnets.0/conv1/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉TransposeTranspose_NCHW_to_NHWC_17GroupNormGroupNorm_17float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_17Conv/encoder/mid_block/resnets.0/conv2/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉Add/encoder/mid_block/resnets.0/AddShape/encoder/mid_block/attentions.0/Shape_3Gather/encoder/mid_block/attentions.0/Gather_3int64indices = 3Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_14int64[1]axes〈1〉Gather/encoder/mid_block/attentions.0/Gather_2int64indices = 2Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_13int64[1]axes〈1〉Gather/encoder/mid_block/attentions.0/Gather_1int64indices = 1Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_12int64[1]axes〈1〉Reshape/encoder/mid_block/attentions.0/Reshapeint64[3]shape〈3〉Shape/encoder/mid_block/attentions.0/Shape_4GatherGatherint64[3]indices〈3〉Gather/encoder/mid_block/attentions.0/Gather_4int64indices = 0Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_11int64[1]axes〈1〉Concat/encoder/mid_block/attentions.0/Concat_5Reshape/encoder/mid_block/attentions.0/group_norm/Reshapeint64[3]shape〈3〉InstanceNormalization/encoder/mid_block/attentions.0/group_norm/InstanceNormalizationfloat16[32]scale〈32〉float16[32]B〈32〉Reshape/encoder/mid_block/attentions.0/group_norm/Reshape_1Mul/encoder/mid_block/attentions.0/group_norm/Mulfloat16[512,1]B〈512×1〉Add/encoder/mid_block/attentions.0/group_norm/Addfloat16[512,1]B〈512×1〉Transpose/encoder/mid_block/attentions.0/Transpose_1MatMul/encoder/mid_block/attentions.0/to_k/MatMulfloat16[512,512]B〈512×512〉Add/encoder/mid_block/attentions.0/to_k/Addfloat16[512]A〈512〉Shape/encoder/mid_block/attentions.0/Shape_5Gather/encoder/mid_block/attentions.0/Gather_5int64indices = 2Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_10int64[1]axes〈1〉Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_9int64[1]axes〈1〉Concat/encoder/mid_block/attentions.0/Concat_4〈…〉Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_4int64[1]axes〈1〉Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_3int64[1]axes〈1〉Concat/encoder/mid_block/attentions.0/Concat_1〈…〉MatMul/encoder/mid_block/attentions.0/to_q/MatMulfloat16[512,512]B〈512×512〉Add/encoder/mid_block/attentions.0/to_q/Addfloat16[512]A〈512〉Reshape/encoder/mid_block/attentions.0/Reshape_1Shape/encoder/mid_block/attentions.0/Shape_6GatherGather_token_1int64[4]indices〈4〉Slice/encoder/mid_block/attentions.0/Sliceint64[1]starts〈1〉int64[1]ends〈1〉Cast/encoder/mid_block/attentions.0/Cast_2Sqrt/encoder/mid_block/attentions.0/SqrtDiv/encoder/mid_block/attentions.0/Div_1float16[1]A〈1〉Sqrt/encoder/mid_block/attentions.0/Sqrt_2Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_6int64[1]axes〈1〉Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_5int64[1]axes〈1〉Concat/encoder/mid_block/attentions.0/Concat_2〈…〉Reshape/encoder/mid_block/attentions.0/Reshape_2Transpose/encoder/mid_block/attentions.0/Transpose_4Mul/encoder/mid_block/attentions.0/Mul_2Transpose/encoder/mid_block/attentions.0/Transpose_2Mul/encoder/mid_block/attentions.0/Mul_1MatMul/encoder/mid_block/attentions.0/MatMulSoftmax/encoder/mid_block/attentions.0/SoftmaxUnsqueeze/encoder/mid_block/attentions.0/Unsqueeze_8int64[1]axes〈1〉Unsqueeze/encoder/mid_block/attentions.0/Unsqueeze_7int64[1]axes〈1〉Concat/encoder/mid_block/attentions.0/Concat_3〈…〉MatMul/encoder/mid_block/attentions.0/to_v/MatMulfloat16[512,512]B〈512×512〉Add/encoder/mid_block/attentions.0/to_v/Addfloat16[512]A〈512〉Reshape/encoder/mid_block/attentions.0/Reshape_3Transpose/encoder/mid_block/attentions.0/Transpose_3MatMul/encoder/mid_block/attentions.0/MatMul_1Transpose/encoder/mid_block/attentions.0/Transpose_5Reshape/encoder/mid_block/attentions.0/Reshape_4MatMul/encoder/mid_block/attentions.0/to_out.0/MatMulfloat16[512,512]B〈512×512〉Add/encoder/mid_block/attentions.0/to_out.0/Addfloat16[512]A〈512〉Transpose/encoder/mid_block/attentions.0/Transpose_6Reshape/encoder/mid_block/attentions.0/Reshape_5Add/encoder/mid_block/attentions.0/AddTransposeTranspose_NCHW_to_NHWC_18GroupNormGroupNorm_18float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_18Conv/encoder/mid_block/resnets.1/conv1/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉TransposeTranspose_NCHW_to_NHWC_19GroupNormGroupNorm_19float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_19Conv/encoder/mid_block/resnets.1/conv2/Convfloat16[512,512,3,3]W〈512×512×3×3〉float16[512]B〈512〉Add/encoder/mid_block/resnets.1/AddTransposeTranspose_NCHW_to_NHWC_20GroupNormGroupNorm_20float32[512]gamma〈512〉float32[512]beta〈512〉TransposeTranspose_NHWC_to_NCHW_20Conv/encoder/conv_out/Convfloat16[8,512,3,3]W〈8×512×3×3〉float16[8]B〈8〉Conv/quant_conv/Convfloat16[8,8,1,1]W〈8×8×1×1〉float16[8]B〈8〉Shape/ShapeGather/Gatherint64[1]indices〈1〉Add/Addint64[1]B〈1〉Div/Divint64[1]B〈1〉Slice/Sliceint64[1]starts〈1〉int64[1]axes〈1〉Shape/Shape_4ConstantOfShape/ConstantOfShapeMul/Mul_1int64[1]B〈1〉Slice/Slice_1int64[1]axes〈1〉Clip/Clipfloat16min = -30float16max = 20Mul/Mul_2float16B = 0.5Exp/ExpCast/RandomNormalLike_input_cast0RandomNormalLike/RandomNormalLikeCast/RandomNormalLike_output_cast0Mul/Mul_3Add/Add_1Castgraph_output_cast0latent_samplefloat32[batch_size,num_channels_latent,height_latent,width_latent]