Skip to content

N
ngraph
Commit 7d34b79a authored by sharathns93's avatar sharathns93 Committed by Robert Kimball
Browse files
Options

TF serialzed graphs (#528) 

* add TF serialzed graphs
parent ec6b3eae
Hide whitespace changes
Inline Side-by-side
Showing with 5774 additions and 0 deletions
test/models/tensorflow/mnist_cnn_maxpool/tf_function_cluster_13[_XlaCompiledKernel=true,_XlaNumConstantArgs=0,_XlaNumResourceArgs=0].v33.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_4",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_176",
"op" : "Parameter",
"outputs" : ["Parameter_176_0"],
"shape" : [ 50, 784 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_175",
"op" : "Parameter",
"outputs" : ["Parameter_175_0"],
"shape" : [10]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_174",
"op" : "Parameter",
"outputs" : ["Parameter_174_0"],
"shape" : [ 1024, 10 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_173",
"op" : "Parameter",
"outputs" : ["Parameter_173_0"],
"shape" : [1024]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_172",
"op" : "Parameter",
"outputs" : ["Parameter_172_0"],
"shape" : [ 3136, 1024 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_171",
"op" : "Parameter",
"outputs" : ["Parameter_171_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_170",
"op" : "Parameter",
"outputs" : ["Parameter_170_0"],
"shape" : [ 5, 5, 32, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_169",
"op" : "Parameter",
"outputs" : ["Parameter_169_0"],
"shape" : [32]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_168",
"op" : "Parameter",
"outputs" : ["Parameter_168_0"],
"shape" : [ 5, 5, 1, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_179",
"op" : "Constant",
"outputs" : ["Constant_179_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_178",
"op" : "Constant",
"outputs" : ["Constant_178_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_177",
"op" : "Constant",
"outputs" : ["Constant_177_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Parameter_176"],
"name" : "Reshape_180",
"op" : "Reshape",
"output_shape" : [ 50, 28, 28, 1 ],
"outputs" : ["Reshape_180_0"]
},
{
"axes" : [0],
"inputs" : ["Parameter_175"],
"name" : "Broadcast_210",
"op" : "Broadcast",
"outputs" : ["Broadcast_210_0"],
"shape" : [ 50, 10 ]
},
{
"axes" : [0],
"inputs" : ["Parameter_173"],
"name" : "Broadcast_205",
"op" : "Broadcast",
"outputs" : ["Broadcast_205_0"],
"shape" : [ 50, 1024 ]
},
{
"axes" : [ 0, 1, 2 ],
"inputs" : ["Parameter_171"],
"name" : "Broadcast_196",
"op" : "Broadcast",
"outputs" : ["Broadcast_196_0"],
"shape" : [ 50, 14, 14, 64 ]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_170"],
"name" : "Reshape_193",
"op" : "Reshape",
"output_shape" : [ 64, 32, 5, 5 ],
"outputs" : ["Reshape_193_0"]
},
{
"axes" : [ 0, 1, 2 ],
"inputs" : ["Parameter_169"],
"name" : "Broadcast_185",
"op" : "Broadcast",
"outputs" : ["Broadcast_185_0"],
"shape" : [ 50, 28, 28, 32 ]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_168"],
"name" : "Reshape_182",
"op" : "Reshape",
"output_shape" : [ 32, 1, 5, 5 ],
"outputs" : ["Reshape_182_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_179"],
"name" : "Broadcast_187",
"op" : "Broadcast",
"outputs" : ["Broadcast_187_0"],
"shape" : [ 50, 28, 28, 32 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_178"],
"name" : "Broadcast_198",
"op" : "Broadcast",
"outputs" : ["Broadcast_198_0"],
"shape" : [ 50, 14, 14, 64 ]
},
{
"axes" : [ 0, 1 ],
"inputs" : ["Constant_177"],
"name" : "Broadcast_207",
"op" : "Broadcast",
"outputs" : ["Broadcast_207_0"],
"shape" : [ 50, 1024 ]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_180"],
"name" : "Reshape_181",
"op" : "Reshape",
"output_shape" : [ 50, 1, 28, 28 ],
"outputs" : ["Reshape_181_0"]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_181", "Reshape_182" ],
"name" : "Convolution_183",
"op" : "Convolution",
"outputs" : ["Convolution_183_0"],
"padding_above" : [ 2, 2 ],
"padding_below" : [ 2, 2 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["Convolution_183"],
"name" : "Reshape_184",
"op" : "Reshape",
"output_shape" : [ 50, 28, 28, 32 ],
"outputs" : ["Reshape_184_0"]
},
{
"inputs" : [ "Reshape_184", "Broadcast_185" ],
"name" : "Add_186",
"op" : "Add",
"outputs" : ["Add_186_0"]
},
{
"inputs" : [ "Broadcast_187", "Add_186" ],
"name" : "Maximum_188",
"op" : "Maximum",
"outputs" : ["Maximum_188_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Maximum_188"],
"name" : "Reshape_189",
"op" : "Reshape",
"output_shape" : [ 50, 32, 28, 28 ],
"outputs" : ["Reshape_189_0"]
},
{
"inputs" : ["Reshape_189"],
"name" : "MaxPool_190",
"op" : "MaxPool",
"outputs" : ["MaxPool_190_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_movement_strides" : [ 2, 2 ],
"window_shape" : [ 2, 2 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["MaxPool_190"],
"name" : "Reshape_191",
"op" : "Reshape",
"output_shape" : [ 50, 14, 14, 32 ],
"outputs" : ["Reshape_191_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_191"],
"name" : "Reshape_192",
"op" : "Reshape",
"output_shape" : [ 50, 32, 14, 14 ],
"outputs" : ["Reshape_192_0"]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_192", "Reshape_193" ],
"name" : "Convolution_194",
"op" : "Convolution",
"outputs" : ["Convolution_194_0"],
"padding_above" : [ 2, 2 ],
"padding_below" : [ 2, 2 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["Convolution_194"],
"name" : "Reshape_195",
"op" : "Reshape",
"output_shape" : [ 50, 14, 14, 64 ],
"outputs" : ["Reshape_195_0"]
},
{
"inputs" : [ "Reshape_195", "Broadcast_196" ],
"name" : "Add_197",
"op" : "Add",
"outputs" : ["Add_197_0"]
},
{
"inputs" : [ "Broadcast_198", "Add_197" ],
"name" : "Maximum_199",
"op" : "Maximum",
"outputs" : ["Maximum_199_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Maximum_199"],
"name" : "Reshape_200",
"op" : "Reshape",
"output_shape" : [ 50, 64, 14, 14 ],
"outputs" : ["Reshape_200_0"]
},
{
"inputs" : ["Reshape_200"],
"name" : "MaxPool_201",
"op" : "MaxPool",
"outputs" : ["MaxPool_201_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_movement_strides" : [ 2, 2 ],
"window_shape" : [ 2, 2 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["MaxPool_201"],
"name" : "Reshape_202",
"op" : "Reshape",
"output_shape" : [ 50, 7, 7, 64 ],
"outputs" : ["Reshape_202_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Reshape_202"],
"name" : "Reshape_203",
"op" : "Reshape",
"output_shape" : [ 50, 3136 ],
"outputs" : ["Reshape_203_0"]
},
{
"inputs" : [ "Reshape_203", "Parameter_172" ],
"name" : "Dot_204",
"op" : "Dot",
"outputs" : ["Dot_204_0"],
"reduction_axes_count" : 1
},
{
"inputs" : [ "Dot_204", "Broadcast_205" ],
"name" : "Add_206",
"op" : "Add",
"outputs" : ["Add_206_0"]
},
{
"inputs" : [ "Broadcast_207", "Add_206" ],
"name" : "Maximum_208",
"op" : "Maximum",
"outputs" : ["Maximum_208_0"]
},
{
"inputs" : [ "Maximum_208", "Parameter_174" ],
"name" : "Dot_209",
"op" : "Dot",
"outputs" : ["Dot_209_0"],
"reduction_axes_count" : 1
},
{
"inputs" : [ "Dot_209", "Broadcast_210" ],
"name" : "Add_211",
"op" : "Add",
"outputs" : ["Add_211_0"]
}
],
"parameters" : [
"Parameter_168", "Parameter_169", "Parameter_170", "Parameter_171",
"Parameter_172", "Parameter_173", "Parameter_174", "Parameter_175",
"Parameter_176"
],
"result" : ["Add_211"]
}]
\ No newline at end of file
test/models/tensorflow/mnist_cnn_maxpool/tf_function_cluster_17[_XlaCompiledKernel=true,_XlaNumConstantArgs=1,_XlaNumResourceArgs=0].v170.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_6",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_323",
"op" : "Parameter",
"outputs" : ["Parameter_323_0"],
"shape" : [ 50, 10 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_322",
"op" : "Parameter",
"outputs" : ["Parameter_322_0"],
"shape" : [ 50, 784 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_321",
"op" : "Parameter",
"outputs" : ["Parameter_321_0"],
"shape" : [10]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_320",
"op" : "Parameter",
"outputs" : ["Parameter_320_0"],
"shape" : [ 1024, 10 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_319",
"op" : "Parameter",
"outputs" : ["Parameter_319_0"],
"shape" : [1024]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_318",
"op" : "Parameter",
"outputs" : ["Parameter_318_0"],
"shape" : [ 3136, 1024 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_317",
"op" : "Parameter",
"outputs" : ["Parameter_317_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_316",
"op" : "Parameter",
"outputs" : ["Parameter_316_0"],
"shape" : [ 5, 5, 32, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_315",
"op" : "Parameter",
"outputs" : ["Parameter_315_0"],
"shape" : [32]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_314",
"op" : "Parameter",
"outputs" : ["Parameter_314_0"],
"shape" : [ 5, 5, 1, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_469",
"op" : "Constant",
"outputs" : ["Constant_469_0"],
"shape" : [],
"value" : ["50"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_326",
"op" : "Constant",
"outputs" : ["Constant_326_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_325",
"op" : "Constant",
"outputs" : ["Constant_325_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_324",
"op" : "Constant",
"outputs" : ["Constant_324_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_441",
"op" : "Constant",
"outputs" : ["Constant_441_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_422",
"op" : "Constant",
"outputs" : ["Constant_422_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_411",
"op" : "Constant",
"outputs" : ["Constant_411_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "int32_t",
"inputs" : [],
"name" : "Constant_372",
"op" : "Constant",
"outputs" : ["Constant_372_0"],
"shape" : [],
"value" : ["1"]
},
{
"element_type" : "int32_t",
"inputs" : [],
"name" : "Constant_365",
"op" : "Constant",
"outputs" : ["Constant_365_0"],
"shape" : [1],
"value" : ["50"]
},
{
"element_type" : "int32_t",
"inputs" : [],
"name" : "Constant_381",
"op" : "Constant",
"outputs" : ["Constant_381_0"],
"shape" : [],
"value" : ["1"]
},
{
"element_type" : "int32_t",
"inputs" : [],
"name" : "Constant_370",
"op" : "Constant",
"outputs" : ["Constant_370_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_361",
"op" : "Constant",
"outputs" : ["Constant_361_0"],
"shape" : [1],
"value" : ["1"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_359",
"op" : "Constant",
"outputs" : ["Constant_359_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Parameter_323"],
"name" : "Reshape_400",
"op" : "Reshape",
"output_shape" : [ 50, 10 ],
"outputs" : ["Reshape_400_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Parameter_322"],
"name" : "Reshape_327",
"op" : "Reshape",
"output_shape" : [ 50, 28, 28, 1 ],
"outputs" : ["Reshape_327_0"]
},
{
"axes" : [0],
"inputs" : ["Parameter_321"],
"name" : "Broadcast_357",
"op" : "Broadcast",
"outputs" : ["Broadcast_357_0"],
"shape" : [ 50, 10 ]
},
{
"input_order" : [ 1, 0 ],
"inputs" : ["Parameter_320"],
"name" : "Reshape_414",
"op" : "Reshape",
"output_shape" : [ 10, 1024 ],
"outputs" : ["Reshape_414_0"]
},
{
"axes" : [0],
"inputs" : ["Parameter_319"],
"name" : "Broadcast_352",
"op" : "Broadcast",
"outputs" : ["Broadcast_352_0"],
"shape" : [ 50, 1024 ]
},
{
"input_order" : [ 1, 0 ],
"inputs" : ["Parameter_318"],
"name" : "Reshape_425",
"op" : "Reshape",
"output_shape" : [ 1024, 3136 ],
"outputs" : ["Reshape_425_0"]
},
{
"axes" : [ 0, 1, 2 ],
"inputs" : ["Parameter_317"],
"name" : "Broadcast_343",
"op" : "Broadcast",
"outputs" : ["Broadcast_343_0"],
"shape" : [ 50, 14, 14, 64 ]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_316"],
"name" : "Reshape_340",
"op" : "Reshape",
"output_shape" : [ 64, 32, 5, 5 ],
"outputs" : ["Reshape_340_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_316"],
"name" : "Reshape_445",
"op" : "Reshape",
"output_shape" : [ 64, 32, 5, 5 ],
"outputs" : ["Reshape_445_0"]
},
{
"axes" : [ 0, 1, 2 ],
"inputs" : ["Parameter_315"],
"name" : "Broadcast_332",
"op" : "Broadcast",
"outputs" : ["Broadcast_332_0"],
"shape" : [ 50, 28, 28, 32 ]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_314"],
"name" : "Reshape_329",
"op" : "Reshape",
"output_shape" : [ 32, 1, 5, 5 ],
"outputs" : ["Reshape_329_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_326"],
"name" : "Broadcast_334",
"op" : "Broadcast",
"outputs" : ["Broadcast_334_0"],
"shape" : [ 50, 28, 28, 32 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_325"],
"name" : "Broadcast_345",
"op" : "Broadcast",
"outputs" : ["Broadcast_345_0"],
"shape" : [ 50, 14, 14, 64 ]
},
{
"axes" : [ 0, 1 ],
"inputs" : ["Constant_324"],
"name" : "Broadcast_354",
"op" : "Broadcast",
"outputs" : ["Broadcast_354_0"],
"shape" : [ 50, 1024 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_441"],
"name" : "Broadcast_442",
"op" : "Broadcast",
"outputs" : ["Broadcast_442_0"],
"shape" : [ 50, 28, 28, 32 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_422"],
"name" : "Broadcast_423",
"op" : "Broadcast",
"outputs" : ["Broadcast_423_0"],
"shape" : [ 50, 14, 14, 64 ]
},
{
"axes" : [ 0, 1 ],
"inputs" : ["Constant_411"],
"name" : "Broadcast_412",
"op" : "Broadcast",
"outputs" : ["Broadcast_412_0"],
"shape" : [ 50, 1024 ]
},
{
"input_order" : [],
"inputs" : ["Constant_372"],
"name" : "Reshape_373",
"op" : "Reshape",
"output_shape" : [1],
"outputs" : ["Reshape_373_0"]
},
{
"inputs" : ["Constant_365"],
"name" : "Product_366",
"op" : "Product",
"outputs" : ["Product_366_0"],
"reduction_axes" : [0]
},
{
"input_order" : [0],
"inputs" : ["Constant_361"],
"name" : "Reshape_362",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_362_0"]
},
{
"axes" : [0],
"inputs" : ["Constant_359"],
"name" : "Broadcast_360",
"op" : "Broadcast",
"outputs" : ["Broadcast_360_0"],
"shape" : [50]
},
{
"inputs" : ["Reshape_400"],
"name" : "Negative_461",
"op" : "Negative",
"outputs" : ["Negative_461_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Reshape_327"],
"name" : "Reshape_456",
"op" : "Reshape",
"output_shape" : [ 1, 50, 28, 28 ],
"outputs" : ["Reshape_456_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_327"],
"name" : "Reshape_328",
"op" : "Reshape",
"output_shape" : [ 50, 1, 28, 28 ],
"outputs" : ["Reshape_328_0"]
},
{
"axes" : [0],
"inputs" : ["Reshape_373"],
"name" : "Broadcast_374",
"op" : "Broadcast",
"outputs" : ["Broadcast_374_0"],
"shape" : [ 1, 1 ]
},
{
"input_order" : [],
"inputs" : ["Product_366"],
"name" : "Reshape_367",
"op" : "Reshape",
"output_shape" : [1],
"outputs" : ["Reshape_367_0"]
},
{
"axes" : [0],
"inputs" : ["Reshape_362"],
"name" : "Broadcast_363",
"op" : "Broadcast",
"outputs" : ["Broadcast_363_0"],
"shape" : [50]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_328", "Reshape_329" ],
"name" : "Convolution_330",
"op" : "Convolution",
"outputs" : ["Convolution_330_0"],
"padding_above" : [ 2, 2 ],
"padding_below" : [ 2, 2 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Broadcast_374"],
"name" : "Reshape_375",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_375_0"]
},
{
"axes" : [0],
"inputs" : ["Reshape_367"],
"name" : "Broadcast_368",
"op" : "Broadcast",
"outputs" : ["Broadcast_368_0"],
"shape" : [ 1, 1 ]
},
{
"inputs" : [ "Broadcast_360", "Broadcast_363" ],
"name" : "Add_364",
"op" : "Add",
"outputs" : ["Add_364_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["Convolution_330"],
"name" : "Reshape_331",
"op" : "Reshape",
"output_shape" : [ 50, 28, 28, 32 ],
"outputs" : ["Reshape_331_0"]
},
{
"inputs" : [ "Reshape_375", "Constant_370" ],
"name" : "Less_376",
"op" : "Less",
"outputs" : ["Less_376_0"]
},
{
"inputs" : ["Reshape_375"],
"name" : "Abs_379",
"op" : "Abs",
"outputs" : ["Abs_379_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Broadcast_368"],
"name" : "Reshape_369",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_369_0"]
},
{
"inputs" : [ "Reshape_331", "Broadcast_332" ],
"name" : "Add_333",
"op" : "Add",
"outputs" : ["Add_333_0"]
},
{
"inputs" : [ "Reshape_369", "Constant_370" ],
"name" : "Less_371",
"op" : "Less",
"outputs" : ["Less_371_0"]
},
{
"inputs" : [ "Reshape_369", "Reshape_375" ],
"name" : "Divide_385",
"op" : "Divide",
"outputs" : ["Divide_385_0"]
},
{
"inputs" : ["Reshape_369"],
"name" : "Abs_378",
"op" : "Abs",
"outputs" : ["Abs_378_0"]
},
{
"inputs" : [ "Broadcast_334", "Add_333" ],
"name" : "Maximum_335",
"op" : "Maximum",
"outputs" : ["Maximum_335_0"]
},
{
"inputs" : [ "Less_371", "Less_376" ],
"name" : "NotEqual_377",
"op" : "NotEqual",
"outputs" : ["NotEqual_377_0"]
},
{
"inputs" : [ "Abs_378", "Abs_379" ],
"name" : "Add_380",
"op" : "Add",
"outputs" : ["Add_380_0"]
},
{
"inputs" : [ "Maximum_335", "Broadcast_442" ],
"name" : "Greater_443",
"op" : "Greater",
"outputs" : ["Greater_443_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Maximum_335"],
"name" : "Reshape_336",
"op" : "Reshape",
"output_shape" : [ 50, 32, 28, 28 ],
"outputs" : ["Reshape_336_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Maximum_335"],
"name" : "Reshape_448",
"op" : "Reshape",
"output_shape" : [ 50, 32, 28, 28 ],
"outputs" : ["Reshape_448_0"]
},
{
"inputs" : [ "Add_380", "Constant_381" ],
"name" : "Subtract_382",
"op" : "Subtract",
"outputs" : ["Subtract_382_0"]
},
{
"inputs" : ["Reshape_336"],
"name" : "MaxPool_337",
"op" : "MaxPool",
"outputs" : ["MaxPool_337_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_movement_strides" : [ 2, 2 ],
"window_shape" : [ 2, 2 ]
},
{
"inputs" : ["Subtract_382"],
"name" : "Negative_383",
"op" : "Negative",
"outputs" : ["Negative_383_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["MaxPool_337"],
"name" : "Reshape_338",
"op" : "Reshape",
"output_shape" : [ 50, 14, 14, 32 ],
"outputs" : ["Reshape_338_0"]
},
{
"inputs" : [ "Negative_383", "Abs_379" ],
"name" : "Divide_384",
"op" : "Divide",
"outputs" : ["Divide_384_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Reshape_338"],
"name" : "Reshape_436",
"op" : "Reshape",
"output_shape" : [ 32, 50, 14, 14 ],
"outputs" : ["Reshape_436_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_338"],
"name" : "Reshape_339",
"op" : "Reshape",
"output_shape" : [ 50, 32, 14, 14 ],
"outputs" : ["Reshape_339_0"]
},
{
"inputs" : [ "NotEqual_377", "Divide_384", "Divide_385" ],
"name" : "Select_386",
"op" : "Select",
"outputs" : ["Select_386_0"]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_339", "Reshape_340" ],
"name" : "Convolution_341",
"op" : "Convolution",
"outputs" : ["Convolution_341_0"],
"padding_above" : [ 2, 2 ],
"padding_below" : [ 2, 2 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"inputs" : ["Select_386"],
"name" : "Convert_387",
"op" : "Convert",
"outputs" : ["Convert_387_0"],
"target_type" : "float"
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["Convolution_341"],
"name" : "Reshape_342",
"op" : "Reshape",
"output_shape" : [ 50, 14, 14, 64 ],
"outputs" : ["Reshape_342_0"]
},
{
"axes" : [0],
"inputs" : ["Convert_387"],
"name" : "Broadcast_388",
"op" : "Broadcast",
"outputs" : ["Broadcast_388_0"],
"shape" : [50]
},
{
"inputs" : [ "Reshape_342", "Broadcast_343" ],
"name" : "Add_344",
"op" : "Add",
"outputs" : ["Add_344_0"]
},
{
"inputs" : [ "Add_364", "Broadcast_388" ],
"name" : "Divide_389",
"op" : "Divide",
"outputs" : ["Divide_389_0"]
},
{
"inputs" : [ "Broadcast_345", "Add_344" ],
"name" : "Maximum_346",
"op" : "Maximum",
"outputs" : ["Maximum_346_0"]
},
{
"input_order" : [0],
"inputs" : ["Divide_389"],
"name" : "Reshape_390",
"op" : "Reshape",
"output_shape" : [50],
"outputs" : ["Reshape_390_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Maximum_346"],
"name" : "Reshape_428",
"op" : "Reshape",
"output_shape" : [ 50, 64, 14, 14 ],
"outputs" : ["Reshape_428_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Maximum_346"],
"name" : "Reshape_347",
"op" : "Reshape",
"output_shape" : [ 50, 64, 14, 14 ],
"outputs" : ["Reshape_347_0"]
},
{
"inputs" : [ "Maximum_346", "Broadcast_423" ],
"name" : "Greater_424",
"op" : "Greater",
"outputs" : ["Greater_424_0"]
},
{
"input_order" : [0],
"inputs" : ["Reshape_390"],
"name" : "Reshape_391",
"op" : "Reshape",
"output_shape" : [ 50, 1 ],
"outputs" : ["Reshape_391_0"]
},
{
"inputs" : ["Reshape_347"],
"name" : "MaxPool_348",
"op" : "MaxPool",
"outputs" : ["MaxPool_348_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_movement_strides" : [ 2, 2 ],
"window_shape" : [ 2, 2 ]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Reshape_391"],
"name" : "Reshape_402",
"op" : "Reshape",
"output_shape" : [50],
"outputs" : ["Reshape_402_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["MaxPool_348"],
"name" : "Reshape_349",
"op" : "Reshape",
"output_shape" : [ 50, 7, 7, 64 ],
"outputs" : ["Reshape_349_0"]
},
{
"axes" : [1],
"inputs" : ["Reshape_402"],
"name" : "Broadcast_403",
"op" : "Broadcast",
"outputs" : ["Broadcast_403_0"],
"shape" : [ 50, 10 ]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Reshape_349"],
"name" : "Reshape_350",
"op" : "Reshape",
"output_shape" : [ 50, 3136 ],
"outputs" : ["Reshape_350_0"]
},
{
"input_order" : [ 1, 0 ],
"inputs" : ["Reshape_350"],
"name" : "Reshape_419",
"op" : "Reshape",
"output_shape" : [ 3136, 50 ],
"outputs" : ["Reshape_419_0"]
},
{
"inputs" : [ "Reshape_350", "Parameter_318" ],
"name" : "Dot_351",
"op" : "Dot",
"outputs" : ["Dot_351_0"],
"reduction_axes_count" : 1
},
{
"inputs" : [ "Dot_351", "Broadcast_352" ],
"name" : "Add_353",
"op" : "Add",
"outputs" : ["Add_353_0"]
},
{
"inputs" : [ "Broadcast_354", "Add_353" ],
"name" : "Maximum_355",
"op" : "Maximum",
"outputs" : ["Maximum_355_0"]
},
{
"inputs" : [ "Maximum_355", "Parameter_320" ],
"name" : "Dot_356",
"op" : "Dot",
"outputs" : ["Dot_356_0"],
"reduction_axes_count" : 1
},
{
"inputs" : [ "Maximum_355", "Broadcast_412" ],
"name" : "Greater_413",
"op" : "Greater",
"outputs" : ["Greater_413_0"]
},
{
"input_order" : [ 1, 0 ],
"inputs" : ["Maximum_355"],
"name" : "Reshape_408",
"op" : "Reshape",
"output_shape" : [ 1024, 50 ],
"outputs" : ["Reshape_408_0"]
},
{
"inputs" : [ "Dot_356", "Broadcast_357" ],
"name" : "Add_358",
"op" : "Add",
"outputs" : ["Add_358_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Add_358"],
"name" : "Reshape_392",
"op" : "Reshape",
"output_shape" : [ 50, 10 ],
"outputs" : ["Reshape_392_0"]
},
{
"inputs" : ["Reshape_392"],
"name" : "Max_393",
"op" : "Max",
"outputs" : ["Max_393_0"],
"reduction_axes" : [1]
},
{
"axes" : [1],
"inputs" : ["Max_393"],
"name" : "Broadcast_394",
"op" : "Broadcast",
"outputs" : ["Broadcast_394_0"],
"shape" : [ 50, 10 ]
},
{
"inputs" : [ "Reshape_392", "Broadcast_394" ],
"name" : "Subtract_395",
"op" : "Subtract",
"outputs" : ["Subtract_395_0"]
},
{
"inputs" : ["Subtract_395"],
"name" : "Exp_396",
"op" : "Exp",
"outputs" : ["Exp_396_0"]
},
{
"inputs" : ["Exp_396"],
"name" : "Sum_397",
"op" : "Sum",
"outputs" : ["Sum_397_0"],
"reduction_axes" : [1]
},
{
"inputs" : ["Sum_397"],
"name" : "Log_462",
"op" : "Log",
"outputs" : ["Log_462_0"]
},
{
"axes" : [1],
"inputs" : ["Sum_397"],
"name" : "Broadcast_398",
"op" : "Broadcast",
"outputs" : ["Broadcast_398_0"],
"shape" : [ 50, 10 ]
},
{
"axes" : [1],
"inputs" : ["Log_462"],
"name" : "Broadcast_463",
"op" : "Broadcast",
"outputs" : ["Broadcast_463_0"],
"shape" : [ 50, 10 ]
},
{
"inputs" : [ "Exp_396", "Broadcast_398" ],
"name" : "Divide_399",
"op" : "Divide",
"outputs" : ["Divide_399_0"]
},
{
"inputs" : [ "Subtract_395", "Broadcast_463" ],
"name" : "Subtract_464",
"op" : "Subtract",
"outputs" : ["Subtract_464_0"]
},
{
"inputs" : [ "Divide_399", "Reshape_400" ],
"name" : "Subtract_401",
"op" : "Subtract",
"outputs" : ["Subtract_401_0"]
},
{
"inputs" : [ "Negative_461", "Subtract_464" ],
"name" : "Multiply_465",
"op" : "Multiply",
"outputs" : ["Multiply_465_0"]
},
{
"inputs" : [ "Broadcast_403", "Subtract_401" ],
"name" : "Multiply_404",
"op" : "Multiply",
"outputs" : ["Multiply_404_0"]
},
{
"inputs" : ["Multiply_465"],
"name" : "Sum_466",
"op" : "Sum",
"outputs" : ["Sum_466_0"],
"reduction_axes" : [1]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Multiply_404"],
"name" : "Reshape_405",
"op" : "Reshape",
"output_shape" : [ 50, 10 ],
"outputs" : ["Reshape_405_0"]
},
{
"input_order" : [0],
"inputs" : ["Sum_466"],
"name" : "Reshape_467",
"op" : "Reshape",
"output_shape" : [50],
"outputs" : ["Reshape_467_0"]
},
{
"inputs" : ["Reshape_405"],
"name" : "Sum_406",
"op" : "Sum",
"outputs" : ["Sum_406_0"],
"reduction_axes" : [0]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Reshape_405"],
"name" : "Reshape_409",
"op" : "Reshape",
"output_shape" : [ 50, 10 ],
"outputs" : ["Reshape_409_0"]
},
{
"inputs" : ["Reshape_467"],
"name" : "Sum_468",
"op" : "Sum",
"outputs" : ["Sum_468_0"],
"reduction_axes" : [0]
},
{
"input_order" : [0],
"inputs" : ["Sum_406"],
"name" : "Reshape_407",
"op" : "Reshape",
"output_shape" : [10],
"outputs" : ["Reshape_407_0"]
},
{
"inputs" : [ "Reshape_408", "Reshape_409" ],
"name" : "Dot_410",
"op" : "Dot",
"outputs" : ["Dot_410_0"],
"reduction_axes_count" : 1
},
{
"inputs" : [ "Reshape_409", "Reshape_414" ],
"name" : "Dot_415",
"op" : "Dot",
"outputs" : ["Dot_415_0"],
"reduction_axes_count" : 1
},
{
"inputs" : [ "Sum_468", "Constant_469" ],
"name" : "Divide_470",
"op" : "Divide",
"outputs" : ["Divide_470_0"]
},
{
"inputs" : [ "Greater_413", "Dot_415", "Broadcast_412" ],
"name" : "Select_416",
"op" : "Select",
"outputs" : ["Select_416_0"]
},
{
"inputs" : ["Select_416"],
"name" : "Sum_417",
"op" : "Sum",
"outputs" : ["Sum_417_0"],
"reduction_axes" : [0]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Select_416"],
"name" : "Reshape_420",
"op" : "Reshape",
"output_shape" : [ 50, 1024 ],
"outputs" : ["Reshape_420_0"]
},
{
"input_order" : [0],
"inputs" : ["Sum_417"],
"name" : "Reshape_418",
"op" : "Reshape",
"output_shape" : [1024],
"outputs" : ["Reshape_418_0"]
},
{
"inputs" : [ "Reshape_420", "Reshape_425" ],
"name" : "Dot_426",
"op" : "Dot",
"outputs" : ["Dot_426_0"],
"reduction_axes_count" : 1
},
{
"inputs" : [ "Reshape_419", "Reshape_420" ],
"name" : "Dot_421",
"op" : "Dot",
"outputs" : ["Dot_421_0"],
"reduction_axes_count" : 1
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Dot_426"],
"name" : "Reshape_427",
"op" : "Reshape",
"output_shape" : [ 50, 7, 7, 64 ],
"outputs" : ["Reshape_427_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_427"],
"name" : "Reshape_429",
"op" : "Reshape",
"output_shape" : [ 50, 64, 7, 7 ],
"outputs" : ["Reshape_429_0"]
},
{
"inputs" : [ "Reshape_428", "Reshape_429" ],
"name" : "MaxPoolBackprop_430",
"op" : "MaxPoolBackprop",
"outputs" : ["MaxPoolBackprop_430_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_movement_strides" : [ 2, 2 ],
"window_shape" : [ 2, 2 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["MaxPoolBackprop_430"],
"name" : "Reshape_431",
"op" : "Reshape",
"output_shape" : [ 50, 14, 14, 64 ],
"outputs" : ["Reshape_431_0"]
},
{
"inputs" : [ "Greater_424", "Reshape_431", "Broadcast_423" ],
"name" : "Select_432",
"op" : "Select",
"outputs" : ["Select_432_0"]
},
{
"inputs" : ["Select_432"],
"name" : "Sum_433",
"op" : "Sum",
"outputs" : ["Sum_433_0"],
"reduction_axes" : [ 0, 1, 2 ]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Select_432"],
"name" : "Reshape_435",
"op" : "Reshape",
"output_shape" : [ 50, 14, 14, 64 ],
"outputs" : ["Reshape_435_0"]
},
{
"input_order" : [0],
"inputs" : ["Sum_433"],
"name" : "Reshape_434",
"op" : "Reshape",
"output_shape" : [64],
"outputs" : ["Reshape_434_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Reshape_435"],
"name" : "Reshape_437",
"op" : "Reshape",
"output_shape" : [ 64, 50, 14, 14 ],
"outputs" : ["Reshape_437_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_435"],
"name" : "Reshape_444",
"op" : "Reshape",
"output_shape" : [ 50, 64, 14, 14 ],
"outputs" : ["Reshape_444_0"]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_436", "Reshape_437" ],
"name" : "Convolution_438",
"op" : "Convolution",
"outputs" : ["Convolution_438_0"],
"padding_above" : [ 2, 2 ],
"padding_below" : [ 2, 2 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_batch_shape" : [ 50, 32, 14, 14 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_445", "Reshape_444" ],
"name" : "ConvolutionBackpropData_446",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_446_0"],
"padding_above_forward" : [ 2, 2 ],
"padding_below_forward" : [ 2, 2 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 1, 1 ]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_438"],
"name" : "Reshape_439",
"op" : "Reshape",
"output_shape" : [ 64, 5, 5, 32 ],
"outputs" : ["Reshape_439_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_446"],
"name" : "Reshape_447",
"op" : "Reshape",
"output_shape" : [ 50, 14, 14, 32 ],
"outputs" : ["Reshape_447_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_439"],
"name" : "Reshape_440",
"op" : "Reshape",
"output_shape" : [ 5, 5, 32, 64 ],
"outputs" : ["Reshape_440_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_447"],
"name" : "Reshape_449",
"op" : "Reshape",
"output_shape" : [ 50, 32, 14, 14 ],
"outputs" : ["Reshape_449_0"]
},
{
"inputs" : [ "Reshape_448", "Reshape_449" ],
"name" : "MaxPoolBackprop_450",
"op" : "MaxPoolBackprop",
"outputs" : ["MaxPoolBackprop_450_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_movement_strides" : [ 2, 2 ],
"window_shape" : [ 2, 2 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["MaxPoolBackprop_450"],
"name" : "Reshape_451",
"op" : "Reshape",
"output_shape" : [ 50, 28, 28, 32 ],
"outputs" : ["Reshape_451_0"]
},
{
"inputs" : [ "Greater_443", "Reshape_451", "Broadcast_442" ],
"name" : "Select_452",
"op" : "Select",
"outputs" : ["Select_452_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Select_452"],
"name" : "Reshape_455",
"op" : "Reshape",
"output_shape" : [ 50, 28, 28, 32 ],
"outputs" : ["Reshape_455_0"]
},
{
"inputs" : ["Select_452"],
"name" : "Sum_453",
"op" : "Sum",
"outputs" : ["Sum_453_0"],
"reduction_axes" : [ 0, 1, 2 ]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Reshape_455"],
"name" : "Reshape_457",
"op" : "Reshape",
"output_shape" : [ 32, 50, 28, 28 ],
"outputs" : ["Reshape_457_0"]
},
{
"input_order" : [0],
"inputs" : ["Sum_453"],
"name" : "Reshape_454",
"op" : "Reshape",
"output_shape" : [32],
"outputs" : ["Reshape_454_0"]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_456", "Reshape_457" ],
"name" : "Convolution_458",
"op" : "Convolution",
"outputs" : ["Convolution_458_0"],
"padding_above" : [ 2, 2 ],
"padding_below" : [ 2, 2 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_458"],
"name" : "Reshape_459",
"op" : "Reshape",
"output_shape" : [ 32, 5, 5, 1 ],
"outputs" : ["Reshape_459_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_459"],
"name" : "Reshape_460",
"op" : "Reshape",
"output_shape" : [ 5, 5, 1, 32 ],
"outputs" : ["Reshape_460_0"]
}
],
"parameters" : [
"Parameter_314", "Parameter_315", "Parameter_316", "Parameter_317",
"Parameter_318", "Parameter_319", "Parameter_320", "Parameter_321",
"Parameter_322", "Parameter_323"
],
"result" : [
"Add_358", "Reshape_407", "Dot_410", "Reshape_418", "Dot_421",
"Reshape_434", "Reshape_440", "Reshape_454", "Reshape_460", "Divide_470"
]
}]
\ No newline at end of file
test/models/tensorflow/mnist_mlp/tf_function_cluster_4[_XlaCompiledKernel=true,_XlaNumConstantArgs=1,_XlaNumResourceArgs=0].v82.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_0",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_3",
"op" : "Parameter",
"outputs" : ["Parameter_3_0"],
"shape" : [ 100, 10 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_2",
"op" : "Parameter",
"outputs" : ["Parameter_2_0"],
"shape" : [ 100, 784 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1",
"op" : "Parameter",
"outputs" : ["Parameter_1_0"],
"shape" : [10]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_0",
"op" : "Parameter",
"outputs" : ["Parameter_0_0"],
"shape" : [ 784, 10 ]
},
{
"element_type" : "int32_t",
"inputs" : [],
"name" : "Constant_17",
"op" : "Constant",
"outputs" : ["Constant_17_0"],
"shape" : [],
"value" : ["1"]
},
{
"element_type" : "int32_t",
"inputs" : [],
"name" : "Constant_10",
"op" : "Constant",
"outputs" : ["Constant_10_0"],
"shape" : [1],
"value" : ["100"]
},
{
"element_type" : "int32_t",
"inputs" : [],
"name" : "Constant_26",
"op" : "Constant",
"outputs" : ["Constant_26_0"],
"shape" : [],
"value" : ["1"]
},
{
"element_type" : "int32_t",
"inputs" : [],
"name" : "Constant_15",
"op" : "Constant",
"outputs" : ["Constant_15_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_6",
"op" : "Constant",
"outputs" : ["Constant_6_0"],
"shape" : [1],
"value" : ["1"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_4",
"op" : "Constant",
"outputs" : ["Constant_4_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Parameter_3"],
"name" : "Reshape_48",
"op" : "Reshape",
"output_shape" : [ 100, 10 ],
"outputs" : ["Reshape_48_0"]
},
{
"input_order" : [ 1, 0 ],
"inputs" : ["Parameter_2"],
"name" : "Reshape_56",
"op" : "Reshape",
"output_shape" : [ 784, 100 ],
"outputs" : ["Reshape_56_0"]
},
{
"axes" : [0],
"inputs" : ["Parameter_1"],
"name" : "Broadcast_38",
"op" : "Broadcast",
"outputs" : ["Broadcast_38_0"],
"shape" : [ 100, 10 ]
},
{
"inputs" : [ "Parameter_2", "Parameter_0" ],
"name" : "Dot_37",
"op" : "Dot",
"outputs" : ["Dot_37_0"],
"reduction_axes_count" : 1
},
{
"input_order" : [],
"inputs" : ["Constant_17"],
"name" : "Reshape_18",
"op" : "Reshape",
"output_shape" : [1],
"outputs" : ["Reshape_18_0"]
},
{
"inputs" : ["Constant_10"],
"name" : "Product_11",
"op" : "Product",
"outputs" : ["Product_11_0"],
"reduction_axes" : [0]
},
{
"input_order" : [0],
"inputs" : ["Constant_6"],
"name" : "Reshape_7",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_7_0"]
},
{
"axes" : [0],
"inputs" : ["Constant_4"],
"name" : "Broadcast_5",
"op" : "Broadcast",
"outputs" : ["Broadcast_5_0"],
"shape" : [100]
},
{
"inputs" : [ "Dot_37", "Broadcast_38" ],
"name" : "Add_39",
"op" : "Add",
"outputs" : ["Add_39_0"]
},
{
"axes" : [0],
"inputs" : ["Reshape_18"],
"name" : "Broadcast_19",
"op" : "Broadcast",
"outputs" : ["Broadcast_19_0"],
"shape" : [ 1, 1 ]
},
{
"input_order" : [],
"inputs" : ["Product_11"],
"name" : "Reshape_12",
"op" : "Reshape",
"output_shape" : [1],
"outputs" : ["Reshape_12_0"]
},
{
"axes" : [0],
"inputs" : ["Reshape_7"],
"name" : "Broadcast_8",
"op" : "Broadcast",
"outputs" : ["Broadcast_8_0"],
"shape" : [100]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Add_39"],
"name" : "Reshape_40",
"op" : "Reshape",
"output_shape" : [ 100, 10 ],
"outputs" : ["Reshape_40_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Broadcast_19"],
"name" : "Reshape_20",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_20_0"]
},
{
"axes" : [0],
"inputs" : ["Reshape_12"],
"name" : "Broadcast_13",
"op" : "Broadcast",
"outputs" : ["Broadcast_13_0"],
"shape" : [ 1, 1 ]
},
{
"inputs" : [ "Broadcast_5", "Broadcast_8" ],
"name" : "Add_9",
"op" : "Add",
"outputs" : ["Add_9_0"]
},
{
"inputs" : ["Reshape_40"],
"name" : "Max_41",
"op" : "Max",
"outputs" : ["Max_41_0"],
"reduction_axes" : [1]
},
{
"inputs" : [ "Reshape_20", "Constant_15" ],
"name" : "Less_21",
"op" : "Less",
"outputs" : ["Less_21_0"]
},
{
"inputs" : ["Reshape_20"],
"name" : "Abs_24",
"op" : "Abs",
"outputs" : ["Abs_24_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Broadcast_13"],
"name" : "Reshape_14",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_14_0"]
},
{
"axes" : [1],
"inputs" : ["Max_41"],
"name" : "Broadcast_42",
"op" : "Broadcast",
"outputs" : ["Broadcast_42_0"],
"shape" : [ 100, 10 ]
},
{
"inputs" : ["Reshape_14"],
"name" : "Abs_23",
"op" : "Abs",
"outputs" : ["Abs_23_0"]
},
{
"inputs" : [ "Reshape_14", "Constant_15" ],
"name" : "Less_16",
"op" : "Less",
"outputs" : ["Less_16_0"]
},
{
"inputs" : [ "Reshape_14", "Reshape_20" ],
"name" : "Divide_30",
"op" : "Divide",
"outputs" : ["Divide_30_0"]
},
{
"inputs" : [ "Reshape_40", "Broadcast_42" ],
"name" : "Subtract_43",
"op" : "Subtract",
"outputs" : ["Subtract_43_0"]
},
{
"inputs" : [ "Abs_23", "Abs_24" ],
"name" : "Add_25",
"op" : "Add",
"outputs" : ["Add_25_0"]
},
{
"inputs" : [ "Less_16", "Less_21" ],
"name" : "NotEqual_22",
"op" : "NotEqual",
"outputs" : ["NotEqual_22_0"]
},
{
"inputs" : ["Subtract_43"],
"name" : "Exp_44",
"op" : "Exp",
"outputs" : ["Exp_44_0"]
},
{
"inputs" : [ "Add_25", "Constant_26" ],
"name" : "Subtract_27",
"op" : "Subtract",
"outputs" : ["Subtract_27_0"]
},
{
"inputs" : ["Exp_44"],
"name" : "Sum_45",
"op" : "Sum",
"outputs" : ["Sum_45_0"],
"reduction_axes" : [1]
},
{
"inputs" : ["Subtract_27"],
"name" : "Negative_28",
"op" : "Negative",
"outputs" : ["Negative_28_0"]
},
{
"axes" : [1],
"inputs" : ["Sum_45"],
"name" : "Broadcast_46",
"op" : "Broadcast",
"outputs" : ["Broadcast_46_0"],
"shape" : [ 100, 10 ]
},
{
"inputs" : [ "Negative_28", "Abs_24" ],
"name" : "Divide_29",
"op" : "Divide",
"outputs" : ["Divide_29_0"]
},
{
"inputs" : [ "Exp_44", "Broadcast_46" ],
"name" : "Divide_47",
"op" : "Divide",
"outputs" : ["Divide_47_0"]
},
{
"inputs" : [ "NotEqual_22", "Divide_29", "Divide_30" ],
"name" : "Select_31",
"op" : "Select",
"outputs" : ["Select_31_0"]
},
{
"inputs" : [ "Divide_47", "Reshape_48" ],
"name" : "Subtract_49",
"op" : "Subtract",
"outputs" : ["Subtract_49_0"]
},
{
"inputs" : ["Select_31"],
"name" : "Convert_32",
"op" : "Convert",
"outputs" : ["Convert_32_0"],
"target_type" : "float"
},
{
"axes" : [0],
"inputs" : ["Convert_32"],
"name" : "Broadcast_33",
"op" : "Broadcast",
"outputs" : ["Broadcast_33_0"],
"shape" : [100]
},
{
"inputs" : [ "Add_9", "Broadcast_33" ],
"name" : "Divide_34",
"op" : "Divide",
"outputs" : ["Divide_34_0"]
},
{
"input_order" : [0],
"inputs" : ["Divide_34"],
"name" : "Reshape_35",
"op" : "Reshape",
"output_shape" : [100],
"outputs" : ["Reshape_35_0"]
},
{
"input_order" : [0],
"inputs" : ["Reshape_35"],
"name" : "Reshape_36",
"op" : "Reshape",
"output_shape" : [ 100, 1 ],
"outputs" : ["Reshape_36_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Reshape_36"],
"name" : "Reshape_50",
"op" : "Reshape",
"output_shape" : [100],
"outputs" : ["Reshape_50_0"]
},
{
"axes" : [1],
"inputs" : ["Reshape_50"],
"name" : "Broadcast_51",
"op" : "Broadcast",
"outputs" : ["Broadcast_51_0"],
"shape" : [ 100, 10 ]
},
{
"inputs" : [ "Broadcast_51", "Subtract_49" ],
"name" : "Multiply_52",
"op" : "Multiply",
"outputs" : ["Multiply_52_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Multiply_52"],
"name" : "Reshape_53",
"op" : "Reshape",
"output_shape" : [ 100, 10 ],
"outputs" : ["Reshape_53_0"]
},
{
"inputs" : ["Reshape_53"],
"name" : "Sum_54",
"op" : "Sum",
"outputs" : ["Sum_54_0"],
"reduction_axes" : [0]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Reshape_53"],
"name" : "Reshape_57",
"op" : "Reshape",
"output_shape" : [ 100, 10 ],
"outputs" : ["Reshape_57_0"]
},
{
"input_order" : [0],
"inputs" : ["Sum_54"],
"name" : "Reshape_55",
"op" : "Reshape",
"output_shape" : [10],
"outputs" : ["Reshape_55_0"]
},
{
"inputs" : [ "Reshape_56", "Reshape_57" ],
"name" : "Dot_58",
"op" : "Dot",
"outputs" : ["Dot_58_0"],
"reduction_axes_count" : 1
}
],
"parameters" : [ "Parameter_0", "Parameter_1", "Parameter_2", "Parameter_3" ],
"result" : [ "Reshape_55", "Dot_58" ]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_12[_XlaCompiledKernel=true,_XlaNumConstantArgs=3,_XlaNumResourceArgs=0].v23.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_19",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1379",
"op" : "Parameter",
"outputs" : ["Parameter_1379_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1378",
"op" : "Parameter",
"outputs" : ["Parameter_1378_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1377",
"op" : "Parameter",
"outputs" : ["Parameter_1377_0"],
"shape" : [ 3, 3, 16, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1376",
"op" : "Parameter",
"outputs" : ["Parameter_1376_0"],
"shape" : []
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_1381",
"op" : "Constant",
"outputs" : ["Constant_1381_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1379"],
"name" : "Reshape_1392",
"op" : "Reshape",
"output_shape" : [ 16, 50, 32, 32 ],
"outputs" : ["Reshape_1392_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1378"],
"name" : "Reshape_1384",
"op" : "Reshape",
"output_shape" : [ 50, 32, 32, 16 ],
"outputs" : ["Reshape_1384_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1378"],
"name" : "Reshape_1380",
"op" : "Reshape",
"output_shape" : [ 50, 32, 32, 16 ],
"outputs" : ["Reshape_1380_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_1377"],
"name" : "Reshape_1386",
"op" : "Reshape",
"output_shape" : [ 16, 16, 3, 3 ],
"outputs" : ["Reshape_1386_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1376"],
"name" : "Broadcast_1390",
"op" : "Broadcast",
"outputs" : ["Broadcast_1390_0"],
"shape" : [ 3, 3, 16, 16 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_1381"],
"name" : "Broadcast_1382",
"op" : "Broadcast",
"outputs" : ["Broadcast_1382_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Reshape_1384"],
"name" : "Reshape_1393",
"op" : "Reshape",
"output_shape" : [ 16, 50, 32, 32 ],
"outputs" : ["Reshape_1393_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_1384"],
"name" : "Reshape_1385",
"op" : "Reshape",
"output_shape" : [ 50, 16, 32, 32 ],
"outputs" : ["Reshape_1385_0"]
},
{
"inputs" : [ "Parameter_1377", "Broadcast_1390" ],
"name" : "Multiply_1391",
"op" : "Multiply",
"outputs" : ["Multiply_1391_0"]
},
{
"inputs" : [ "Parameter_1379", "Broadcast_1382" ],
"name" : "Greater_1383",
"op" : "Greater",
"outputs" : ["Greater_1383_0"]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_1392", "Reshape_1393" ],
"name" : "Convolution_1394",
"op" : "Convolution",
"outputs" : ["Convolution_1394_0"],
"padding_above" : [ 1, 1 ],
"padding_below" : [ 1, 1 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_batch_shape" : [ 50, 16, 32, 32 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_1386", "Reshape_1385" ],
"name" : "ConvolutionBackpropData_1387",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_1387_0"],
"padding_above_forward" : [ 1, 1 ],
"padding_below_forward" : [ 1, 1 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 1, 1 ]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_1394"],
"name" : "Reshape_1395",
"op" : "Reshape",
"output_shape" : [ 16, 3, 3, 16 ],
"outputs" : ["Reshape_1395_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_1387"],
"name" : "Reshape_1388",
"op" : "Reshape",
"output_shape" : [ 50, 32, 32, 16 ],
"outputs" : ["Reshape_1388_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_1395"],
"name" : "Reshape_1396",
"op" : "Reshape",
"output_shape" : [ 3, 3, 16, 16 ],
"outputs" : ["Reshape_1396_0"]
},
{
"inputs" : [ "Greater_1383", "Reshape_1388", "Broadcast_1382" ],
"name" : "Select_1389",
"op" : "Select",
"outputs" : ["Select_1389_0"]
},
{
"inputs" : [ "Multiply_1391", "Reshape_1396" ],
"name" : "Add_1397",
"op" : "Add",
"outputs" : ["Add_1397_0"]
}
],
"parameters" : [
"Parameter_1376", "Parameter_1377", "Parameter_1378", "Parameter_1379"
],
"result" : [ "Reshape_1380", "Select_1389", "Add_1397" ]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_16[_XlaCompiledKernel=true,_XlaNumConstantArgs=6,_XlaNumResourceArgs=0].v42.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_16",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1208",
"op" : "Parameter",
"outputs" : ["Parameter_1208_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1207",
"op" : "Parameter",
"outputs" : ["Parameter_1207_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1206",
"op" : "Parameter",
"outputs" : ["Parameter_1206_0"],
"shape" : [ 50, 34, 34, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1205",
"op" : "Parameter",
"outputs" : ["Parameter_1205_0"],
"shape" : [ 1, 1, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1204",
"op" : "Parameter",
"outputs" : ["Parameter_1204_0"],
"shape" : [ 1, 1, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1203",
"op" : "Parameter",
"outputs" : ["Parameter_1203_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1202",
"op" : "Parameter",
"outputs" : ["Parameter_1202_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1201",
"op" : "Parameter",
"outputs" : ["Parameter_1201_0"],
"shape" : [ 3, 3, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1200",
"op" : "Parameter",
"outputs" : ["Parameter_1200_0"],
"shape" : []
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_1209",
"op" : "Constant",
"outputs" : ["Constant_1209_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1207"],
"name" : "Reshape_1226",
"op" : "Reshape",
"output_shape" : [ 32, 50, 16, 16 ],
"outputs" : ["Reshape_1226_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Parameter_1207"],
"name" : "Reshape_1216",
"op" : "Reshape",
"output_shape" : [ 50, 32, 16, 16 ],
"outputs" : ["Reshape_1216_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1206"],
"name" : "Reshape_1225",
"op" : "Reshape",
"output_shape" : [ 16, 50, 34, 34 ],
"outputs" : ["Reshape_1225_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_1204"],
"name" : "Reshape_1213",
"op" : "Reshape",
"output_shape" : [ 32, 16, 1, 1 ],
"outputs" : ["Reshape_1213_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1203"],
"name" : "Reshape_1231",
"op" : "Reshape",
"output_shape" : [ 16, 50, 32, 32 ],
"outputs" : ["Reshape_1231_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1202"],
"name" : "Reshape_1232",
"op" : "Reshape",
"output_shape" : [ 32, 50, 16, 16 ],
"outputs" : ["Reshape_1232_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Parameter_1202"],
"name" : "Reshape_1212",
"op" : "Reshape",
"output_shape" : [ 50, 32, 16, 16 ],
"outputs" : ["Reshape_1212_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_1201"],
"name" : "Reshape_1217",
"op" : "Reshape",
"output_shape" : [ 32, 16, 3, 3 ],
"outputs" : ["Reshape_1217_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1200"],
"name" : "Broadcast_1223",
"op" : "Broadcast",
"outputs" : ["Broadcast_1223_0"],
"shape" : [ 3, 3, 16, 32 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_1209"],
"name" : "Broadcast_1210",
"op" : "Broadcast",
"outputs" : ["Broadcast_1210_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_1225", "Reshape_1226" ],
"name" : "Convolution_1227",
"op" : "Convolution",
"outputs" : ["Convolution_1227_0"],
"padding_above" : [ -1, -1 ],
"padding_below" : [ 0, 0 ],
"window_dilation_strides" : [ 2, 2 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_1231", "Reshape_1232" ],
"name" : "Convolution_1233",
"op" : "Convolution",
"outputs" : ["Convolution_1233_0"],
"padding_above" : [ -1, -1 ],
"padding_below" : [ 0, 0 ],
"window_dilation_strides" : [ 2, 2 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_batch_shape" : [ 50, 16, 32, 32 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_1213", "Reshape_1212" ],
"name" : "ConvolutionBackpropData_1214",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_1214_0"],
"padding_above_forward" : [ 0, 0 ],
"padding_below_forward" : [ 0, 0 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 2, 2 ]
},
{
"data_batch_shape" : [ 50, 16, 34, 34 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_1217", "Reshape_1216" ],
"name" : "ConvolutionBackpropData_1218",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_1218_0"],
"padding_above_forward" : [ 0, 0 ],
"padding_below_forward" : [ 0, 0 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 2, 2 ]
},
{
"inputs" : [ "Parameter_1201", "Broadcast_1223" ],
"name" : "Multiply_1224",
"op" : "Multiply",
"outputs" : ["Multiply_1224_0"]
},
{
"inputs" : [ "Parameter_1208", "Broadcast_1210" ],
"name" : "Greater_1211",
"op" : "Greater",
"outputs" : ["Greater_1211_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_1227"],
"name" : "Reshape_1228",
"op" : "Reshape",
"output_shape" : [ 32, 3, 3, 16 ],
"outputs" : ["Reshape_1228_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_1233"],
"name" : "Reshape_1234",
"op" : "Reshape",
"output_shape" : [ 32, 1, 1, 16 ],
"outputs" : ["Reshape_1234_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_1214"],
"name" : "Reshape_1215",
"op" : "Reshape",
"output_shape" : [ 50, 32, 32, 16 ],
"outputs" : ["Reshape_1215_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_1218"],
"name" : "Reshape_1219",
"op" : "Reshape",
"output_shape" : [ 50, 34, 34, 16 ],
"outputs" : ["Reshape_1219_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_1228"],
"name" : "Reshape_1229",
"op" : "Reshape",
"output_shape" : [ 3, 3, 16, 32 ],
"outputs" : ["Reshape_1229_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_1234"],
"name" : "Reshape_1235",
"op" : "Reshape",
"output_shape" : [ 1, 1, 16, 32 ],
"outputs" : ["Reshape_1235_0"]
},
{
"inputs" : ["Reshape_1219"],
"lower_bounds" : [ 0, 1, 1, 0 ],
"name" : "Slice_1220",
"op" : "Slice",
"outputs" : ["Slice_1220_0"],
"strides" : [ 1, 1, 1, 1 ],
"upper_bounds" : [ 50, 33, 33, 16 ]
},
{
"inputs" : [ "Multiply_1224", "Reshape_1229" ],
"name" : "Add_1230",
"op" : "Add",
"outputs" : ["Add_1230_0"]
},
{
"inputs" : [ "Parameter_1205", "Reshape_1235" ],
"name" : "Add_1236",
"op" : "Add",
"outputs" : ["Add_1236_0"]
},
{
"inputs" : [ "Reshape_1215", "Slice_1220" ],
"name" : "Add_1221",
"op" : "Add",
"outputs" : ["Add_1221_0"]
},
{
"inputs" : [ "Greater_1211", "Add_1221", "Broadcast_1210" ],
"name" : "Select_1222",
"op" : "Select",
"outputs" : ["Select_1222_0"]
}
],
"parameters" : [
"Parameter_1200", "Parameter_1201", "Parameter_1202", "Parameter_1203",
"Parameter_1204", "Parameter_1205", "Parameter_1206", "Parameter_1207",
"Parameter_1208"
],
"result" : [ "Select_1222", "Add_1230", "Add_1236" ]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_20[_XlaCompiledKernel=true,_XlaNumConstantArgs=3,_XlaNumResourceArgs=0].v23.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_13",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1044",
"op" : "Parameter",
"outputs" : ["Parameter_1044_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1043",
"op" : "Parameter",
"outputs" : ["Parameter_1043_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1042",
"op" : "Parameter",
"outputs" : ["Parameter_1042_0"],
"shape" : [ 3, 3, 32, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1041",
"op" : "Parameter",
"outputs" : ["Parameter_1041_0"],
"shape" : []
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_1046",
"op" : "Constant",
"outputs" : ["Constant_1046_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1044"],
"name" : "Reshape_1057",
"op" : "Reshape",
"output_shape" : [ 32, 50, 16, 16 ],
"outputs" : ["Reshape_1057_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1043"],
"name" : "Reshape_1045",
"op" : "Reshape",
"output_shape" : [ 50, 16, 16, 32 ],
"outputs" : ["Reshape_1045_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1043"],
"name" : "Reshape_1049",
"op" : "Reshape",
"output_shape" : [ 50, 16, 16, 32 ],
"outputs" : ["Reshape_1049_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_1042"],
"name" : "Reshape_1051",
"op" : "Reshape",
"output_shape" : [ 32, 32, 3, 3 ],
"outputs" : ["Reshape_1051_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1041"],
"name" : "Broadcast_1055",
"op" : "Broadcast",
"outputs" : ["Broadcast_1055_0"],
"shape" : [ 3, 3, 32, 32 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_1046"],
"name" : "Broadcast_1047",
"op" : "Broadcast",
"outputs" : ["Broadcast_1047_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_1049"],
"name" : "Reshape_1050",
"op" : "Reshape",
"output_shape" : [ 50, 32, 16, 16 ],
"outputs" : ["Reshape_1050_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Reshape_1049"],
"name" : "Reshape_1058",
"op" : "Reshape",
"output_shape" : [ 32, 50, 16, 16 ],
"outputs" : ["Reshape_1058_0"]
},
{
"inputs" : [ "Parameter_1042", "Broadcast_1055" ],
"name" : "Multiply_1056",
"op" : "Multiply",
"outputs" : ["Multiply_1056_0"]
},
{
"inputs" : [ "Parameter_1044", "Broadcast_1047" ],
"name" : "Greater_1048",
"op" : "Greater",
"outputs" : ["Greater_1048_0"]
},
{
"data_batch_shape" : [ 50, 32, 16, 16 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_1051", "Reshape_1050" ],
"name" : "ConvolutionBackpropData_1052",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_1052_0"],
"padding_above_forward" : [ 1, 1 ],
"padding_below_forward" : [ 1, 1 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 1, 1 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_1057", "Reshape_1058" ],
"name" : "Convolution_1059",
"op" : "Convolution",
"outputs" : ["Convolution_1059_0"],
"padding_above" : [ 1, 1 ],
"padding_below" : [ 1, 1 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_1052"],
"name" : "Reshape_1053",
"op" : "Reshape",
"output_shape" : [ 50, 16, 16, 32 ],
"outputs" : ["Reshape_1053_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_1059"],
"name" : "Reshape_1060",
"op" : "Reshape",
"output_shape" : [ 32, 3, 3, 32 ],
"outputs" : ["Reshape_1060_0"]
},
{
"inputs" : [ "Greater_1048", "Reshape_1053", "Broadcast_1047" ],
"name" : "Select_1054",
"op" : "Select",
"outputs" : ["Select_1054_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_1060"],
"name" : "Reshape_1061",
"op" : "Reshape",
"output_shape" : [ 3, 3, 32, 32 ],
"outputs" : ["Reshape_1061_0"]
},
{
"inputs" : [ "Multiply_1056", "Reshape_1061" ],
"name" : "Add_1062",
"op" : "Add",
"outputs" : ["Add_1062_0"]
}
],
"parameters" : [
"Parameter_1041", "Parameter_1042", "Parameter_1043", "Parameter_1044"
],
"result" : [ "Reshape_1045", "Select_1054", "Add_1062" ]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_22[_XlaCompiledKernel=true,_XlaNumConstantArgs=4,_XlaNumResourceArgs=0].v24.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_9",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_763",
"op" : "Parameter",
"outputs" : ["Parameter_763_0"],
"shape" : [ 3, 3, 64, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_762",
"op" : "Parameter",
"outputs" : ["Parameter_762_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_761",
"op" : "Parameter",
"outputs" : ["Parameter_761_0"],
"shape" : [ 1, 1, 32, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_760",
"op" : "Parameter",
"outputs" : ["Parameter_760_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_775",
"op" : "Constant",
"outputs" : ["Constant_775_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_763"],
"name" : "Reshape_779",
"op" : "Reshape",
"output_shape" : [ 64, 64, 3, 3 ],
"outputs" : ["Reshape_779_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_762"],
"name" : "Reshape_770",
"op" : "Reshape",
"output_shape" : [ 64, 50, 8, 8 ],
"outputs" : ["Reshape_770_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_761"],
"name" : "Reshape_766",
"op" : "Reshape",
"output_shape" : [ 64, 32, 1, 1 ],
"outputs" : ["Reshape_766_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_760"],
"name" : "Reshape_764",
"op" : "Reshape",
"output_shape" : [ 50, 8, 8, 64 ],
"outputs" : ["Reshape_764_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_760"],
"name" : "Reshape_769",
"op" : "Reshape",
"output_shape" : [ 50, 8, 8, 64 ],
"outputs" : ["Reshape_769_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_775"],
"name" : "Broadcast_776",
"op" : "Broadcast",
"outputs" : ["Broadcast_776_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_764"],
"name" : "Reshape_765",
"op" : "Reshape",
"output_shape" : [ 50, 64, 8, 8 ],
"outputs" : ["Reshape_765_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Reshape_769"],
"name" : "Reshape_771",
"op" : "Reshape",
"output_shape" : [ 64, 50, 8, 8 ],
"outputs" : ["Reshape_771_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_769"],
"name" : "Reshape_778",
"op" : "Reshape",
"output_shape" : [ 50, 64, 8, 8 ],
"outputs" : ["Reshape_778_0"]
},
{
"inputs" : [ "Parameter_762", "Broadcast_776" ],
"name" : "Greater_777",
"op" : "Greater",
"outputs" : ["Greater_777_0"]
},
{
"data_batch_shape" : [ 50, 32, 16, 16 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_766", "Reshape_765" ],
"name" : "ConvolutionBackpropData_767",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_767_0"],
"padding_above_forward" : [ 0, 0 ],
"padding_below_forward" : [ 0, 0 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 2, 2 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_770", "Reshape_771" ],
"name" : "Convolution_772",
"op" : "Convolution",
"outputs" : ["Convolution_772_0"],
"padding_above" : [ 1, 1 ],
"padding_below" : [ 1, 1 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_batch_shape" : [ 50, 64, 8, 8 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_779", "Reshape_778" ],
"name" : "ConvolutionBackpropData_780",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_780_0"],
"padding_above_forward" : [ 1, 1 ],
"padding_below_forward" : [ 1, 1 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 1, 1 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_767"],
"name" : "Reshape_768",
"op" : "Reshape",
"output_shape" : [ 50, 16, 16, 32 ],
"outputs" : ["Reshape_768_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_772"],
"name" : "Reshape_773",
"op" : "Reshape",
"output_shape" : [ 64, 3, 3, 64 ],
"outputs" : ["Reshape_773_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_780"],
"name" : "Reshape_781",
"op" : "Reshape",
"output_shape" : [ 50, 8, 8, 64 ],
"outputs" : ["Reshape_781_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_773"],
"name" : "Reshape_774",
"op" : "Reshape",
"output_shape" : [ 3, 3, 64, 64 ],
"outputs" : ["Reshape_774_0"]
},
{
"inputs" : [ "Greater_777", "Reshape_781", "Broadcast_776" ],
"name" : "Select_782",
"op" : "Select",
"outputs" : ["Select_782_0"]
}
],
"parameters" :
[ "Parameter_760", "Parameter_761", "Parameter_762", "Parameter_763" ],
"result" : [ "Reshape_764", "Reshape_768", "Reshape_774", "Select_782" ]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_23[_XlaCompiledKernel=true,_XlaNumConstantArgs=1,_XlaNumResourceArgs=0].v296.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_8",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_507",
"op" : "Parameter",
"outputs" : ["Parameter_507_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_506",
"op" : "Parameter",
"outputs" : ["Parameter_506_0"],
"shape" : [ 50, 10 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_505",
"op" : "Parameter",
"outputs" : ["Parameter_505_0"],
"shape" : [ 3, 3, 3, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_504",
"op" : "Parameter",
"outputs" : ["Parameter_504_0"],
"shape" : [16]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_503",
"op" : "Parameter",
"outputs" : ["Parameter_503_0"],
"shape" : [16]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_502",
"op" : "Parameter",
"outputs" : ["Parameter_502_0"],
"shape" : [ 1, 1, 16, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_501",
"op" : "Parameter",
"outputs" : ["Parameter_501_0"],
"shape" : [ 3, 3, 16, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_500",
"op" : "Parameter",
"outputs" : ["Parameter_500_0"],
"shape" : [16]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_499",
"op" : "Parameter",
"outputs" : ["Parameter_499_0"],
"shape" : [16]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_498",
"op" : "Parameter",
"outputs" : ["Parameter_498_0"],
"shape" : [ 3, 3, 16, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_497",
"op" : "Parameter",
"outputs" : ["Parameter_497_0"],
"shape" : [16]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_496",
"op" : "Parameter",
"outputs" : ["Parameter_496_0"],
"shape" : [16]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_495",
"op" : "Parameter",
"outputs" : ["Parameter_495_0"],
"shape" : [ 1, 1, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_494",
"op" : "Parameter",
"outputs" : ["Parameter_494_0"],
"shape" : [ 3, 3, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_493",
"op" : "Parameter",
"outputs" : ["Parameter_493_0"],
"shape" : [32]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_492",
"op" : "Parameter",
"outputs" : ["Parameter_492_0"],
"shape" : [32]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_491",
"op" : "Parameter",
"outputs" : ["Parameter_491_0"],
"shape" : [ 3, 3, 32, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_490",
"op" : "Parameter",
"outputs" : ["Parameter_490_0"],
"shape" : [32]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_489",
"op" : "Parameter",
"outputs" : ["Parameter_489_0"],
"shape" : [32]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_488",
"op" : "Parameter",
"outputs" : ["Parameter_488_0"],
"shape" : [ 1, 1, 32, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_487",
"op" : "Parameter",
"outputs" : ["Parameter_487_0"],
"shape" : [ 3, 3, 32, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_486",
"op" : "Parameter",
"outputs" : ["Parameter_486_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_485",
"op" : "Parameter",
"outputs" : ["Parameter_485_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_484",
"op" : "Parameter",
"outputs" : ["Parameter_484_0"],
"shape" : [ 3, 3, 64, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_483",
"op" : "Parameter",
"outputs" : ["Parameter_483_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_482",
"op" : "Parameter",
"outputs" : ["Parameter_482_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_481",
"op" : "Parameter",
"outputs" : ["Parameter_481_0"],
"shape" : [ 64, 10 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_480",
"op" : "Parameter",
"outputs" : ["Parameter_480_0"],
"shape" : [10]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_514",
"op" : "Constant",
"outputs" : ["Constant_514_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_509",
"op" : "Constant",
"outputs" : ["Constant_509_0"],
"shape" : [],
"value" : ["1"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_508",
"op" : "Constant",
"outputs" : ["Constant_508_0"],
"shape" : [],
"value" : ["1"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_540",
"op" : "Constant",
"outputs" : ["Constant_540_0"],
"shape" : [],
"value" : ["1"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_518",
"op" : "Constant",
"outputs" : ["Constant_518_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_711",
"op" : "Constant",
"outputs" : ["Constant_711_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_706",
"op" : "Constant",
"outputs" : ["Constant_706_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_701",
"op" : "Constant",
"outputs" : ["Constant_701_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_696",
"op" : "Constant",
"outputs" : ["Constant_696_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_691",
"op" : "Constant",
"outputs" : ["Constant_691_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_686",
"op" : "Constant",
"outputs" : ["Constant_686_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_681",
"op" : "Constant",
"outputs" : ["Constant_681_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_676",
"op" : "Constant",
"outputs" : ["Constant_676_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_671",
"op" : "Constant",
"outputs" : ["Constant_671_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_666",
"op" : "Constant",
"outputs" : ["Constant_666_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_661",
"op" : "Constant",
"outputs" : ["Constant_661_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_656",
"op" : "Constant",
"outputs" : ["Constant_656_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_651",
"op" : "Constant",
"outputs" : ["Constant_651_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_646",
"op" : "Constant",
"outputs" : ["Constant_646_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_641",
"op" : "Constant",
"outputs" : ["Constant_641_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_636",
"op" : "Constant",
"outputs" : ["Constant_636_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_631",
"op" : "Constant",
"outputs" : ["Constant_631_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_626",
"op" : "Constant",
"outputs" : ["Constant_626_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_621",
"op" : "Constant",
"outputs" : ["Constant_621_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_616",
"op" : "Constant",
"outputs" : ["Constant_616_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_611",
"op" : "Constant",
"outputs" : ["Constant_611_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_606",
"op" : "Constant",
"outputs" : ["Constant_606_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_601",
"op" : "Constant",
"outputs" : ["Constant_601_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_596",
"op" : "Constant",
"outputs" : ["Constant_596_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_591",
"op" : "Constant",
"outputs" : ["Constant_591_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_587",
"op" : "Constant",
"outputs" : ["Constant_587_0"],
"shape" : [],
"value" : ["2"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_584",
"op" : "Constant",
"outputs" : ["Constant_584_0"],
"shape" : [],
"value" : ["0.0002"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_574",
"op" : "Constant",
"outputs" : ["Constant_574_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_550",
"op" : "Constant",
"outputs" : ["Constant_550_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_548",
"op" : "Constant",
"outputs" : ["Constant_548_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Parameter_506"],
"name" : "Reshape_516",
"op" : "Reshape",
"output_shape" : [ 50, 10 ],
"outputs" : ["Reshape_516_0"]
},
{
"inputs" : [ "Parameter_505", "Parameter_505" ],
"name" : "Multiply_585",
"op" : "Multiply",
"outputs" : ["Multiply_585_0"]
},
{
"inputs" : [ "Parameter_504", "Parameter_504" ],
"name" : "Multiply_589",
"op" : "Multiply",
"outputs" : ["Multiply_589_0"]
},
{
"inputs" : [ "Parameter_503", "Parameter_503" ],
"name" : "Multiply_594",
"op" : "Multiply",
"outputs" : ["Multiply_594_0"]
},
{
"inputs" : [ "Parameter_502", "Parameter_502" ],
"name" : "Multiply_599",
"op" : "Multiply",
"outputs" : ["Multiply_599_0"]
},
{
"inputs" : [ "Parameter_501", "Parameter_501" ],
"name" : "Multiply_604",
"op" : "Multiply",
"outputs" : ["Multiply_604_0"]
},
{
"inputs" : [ "Parameter_500", "Parameter_500" ],
"name" : "Multiply_609",
"op" : "Multiply",
"outputs" : ["Multiply_609_0"]
},
{
"inputs" : [ "Parameter_499", "Parameter_499" ],
"name" : "Multiply_614",
"op" : "Multiply",
"outputs" : ["Multiply_614_0"]
},
{
"inputs" : [ "Parameter_498", "Parameter_498" ],
"name" : "Multiply_619",
"op" : "Multiply",
"outputs" : ["Multiply_619_0"]
},
{
"inputs" : [ "Parameter_497", "Parameter_497" ],
"name" : "Multiply_624",
"op" : "Multiply",
"outputs" : ["Multiply_624_0"]
},
{
"inputs" : [ "Parameter_496", "Parameter_496" ],
"name" : "Multiply_629",
"op" : "Multiply",
"outputs" : ["Multiply_629_0"]
},
{
"inputs" : [ "Parameter_495", "Parameter_495" ],
"name" : "Multiply_634",
"op" : "Multiply",
"outputs" : ["Multiply_634_0"]
},
{
"inputs" : [ "Parameter_494", "Parameter_494" ],
"name" : "Multiply_639",
"op" : "Multiply",
"outputs" : ["Multiply_639_0"]
},
{
"inputs" : [ "Parameter_493", "Parameter_493" ],
"name" : "Multiply_644",
"op" : "Multiply",
"outputs" : ["Multiply_644_0"]
},
{
"inputs" : [ "Parameter_492", "Parameter_492" ],
"name" : "Multiply_649",
"op" : "Multiply",
"outputs" : ["Multiply_649_0"]
},
{
"inputs" : [ "Parameter_491", "Parameter_491" ],
"name" : "Multiply_654",
"op" : "Multiply",
"outputs" : ["Multiply_654_0"]
},
{
"inputs" : [ "Parameter_490", "Parameter_490" ],
"name" : "Multiply_659",
"op" : "Multiply",
"outputs" : ["Multiply_659_0"]
},
{
"inputs" : [ "Parameter_489", "Parameter_489" ],
"name" : "Multiply_664",
"op" : "Multiply",
"outputs" : ["Multiply_664_0"]
},
{
"inputs" : [ "Parameter_488", "Parameter_488" ],
"name" : "Multiply_669",
"op" : "Multiply",
"outputs" : ["Multiply_669_0"]
},
{
"inputs" : [ "Parameter_487", "Parameter_487" ],
"name" : "Multiply_674",
"op" : "Multiply",
"outputs" : ["Multiply_674_0"]
},
{
"inputs" : [ "Parameter_486", "Parameter_486" ],
"name" : "Multiply_679",
"op" : "Multiply",
"outputs" : ["Multiply_679_0"]
},
{
"inputs" : [ "Parameter_485", "Parameter_485" ],
"name" : "Multiply_684",
"op" : "Multiply",
"outputs" : ["Multiply_684_0"]
},
{
"inputs" : [ "Parameter_484", "Parameter_484" ],
"name" : "Multiply_689",
"op" : "Multiply",
"outputs" : ["Multiply_689_0"]
},
{
"inputs" : [ "Parameter_483", "Parameter_483" ],
"name" : "Multiply_694",
"op" : "Multiply",
"outputs" : ["Multiply_694_0"]
},
{
"inputs" : [ "Parameter_482", "Parameter_482" ],
"name" : "Multiply_699",
"op" : "Multiply",
"outputs" : ["Multiply_699_0"]
},
{
"input_order" : [ 1, 0 ],
"inputs" : ["Parameter_481"],
"name" : "Reshape_577",
"op" : "Reshape",
"output_shape" : [ 10, 64 ],
"outputs" : ["Reshape_577_0"]
},
{
"inputs" : [ "Parameter_481", "Parameter_481" ],
"name" : "Multiply_704",
"op" : "Multiply",
"outputs" : ["Multiply_704_0"]
},
{
"axes" : [0],
"inputs" : ["Parameter_480"],
"name" : "Broadcast_526",
"op" : "Broadcast",
"outputs" : ["Broadcast_526_0"],
"shape" : [ 50, 10 ]
},
{
"inputs" : [ "Parameter_480", "Parameter_480" ],
"name" : "Multiply_709",
"op" : "Multiply",
"outputs" : ["Multiply_709_0"]
},
{
"axes" : [0],
"inputs" : ["Constant_509"],
"name" : "Broadcast_510",
"op" : "Broadcast",
"outputs" : ["Broadcast_510_0"],
"shape" : [50]
},
{
"axes" : [0],
"inputs" : ["Constant_508"],
"name" : "Broadcast_511",
"op" : "Broadcast",
"outputs" : ["Broadcast_511_0"],
"shape" : [50]
},
{
"axes" : [0],
"inputs" : ["Constant_540"],
"name" : "Broadcast_541",
"op" : "Broadcast",
"outputs" : ["Broadcast_541_0"],
"shape" : [50]
},
{
"axes" : [0],
"inputs" : ["Constant_540"],
"name" : "Broadcast_559",
"op" : "Broadcast",
"outputs" : ["Broadcast_559_0"],
"shape" : [50]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_518"],
"name" : "Broadcast_519",
"op" : "Broadcast",
"outputs" : ["Broadcast_519_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_574"],
"name" : "Broadcast_575",
"op" : "Broadcast",
"outputs" : ["Broadcast_575_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"axes" : [0],
"inputs" : ["Constant_548"],
"name" : "Broadcast_549",
"op" : "Broadcast",
"outputs" : ["Broadcast_549_0"],
"shape" : [50]
},
{
"inputs" : ["Reshape_516"],
"name" : "Negative_517",
"op" : "Negative",
"outputs" : ["Negative_517_0"]
},
{
"inputs" : ["Multiply_585"],
"name" : "Sum_586",
"op" : "Sum",
"outputs" : ["Sum_586_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_589"],
"name" : "Sum_590",
"op" : "Sum",
"outputs" : ["Sum_590_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_594"],
"name" : "Sum_595",
"op" : "Sum",
"outputs" : ["Sum_595_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_599"],
"name" : "Sum_600",
"op" : "Sum",
"outputs" : ["Sum_600_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_604"],
"name" : "Sum_605",
"op" : "Sum",
"outputs" : ["Sum_605_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_609"],
"name" : "Sum_610",
"op" : "Sum",
"outputs" : ["Sum_610_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_614"],
"name" : "Sum_615",
"op" : "Sum",
"outputs" : ["Sum_615_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_619"],
"name" : "Sum_620",
"op" : "Sum",
"outputs" : ["Sum_620_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_624"],
"name" : "Sum_625",
"op" : "Sum",
"outputs" : ["Sum_625_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_629"],
"name" : "Sum_630",
"op" : "Sum",
"outputs" : ["Sum_630_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_634"],
"name" : "Sum_635",
"op" : "Sum",
"outputs" : ["Sum_635_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_639"],
"name" : "Sum_640",
"op" : "Sum",
"outputs" : ["Sum_640_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_644"],
"name" : "Sum_645",
"op" : "Sum",
"outputs" : ["Sum_645_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_649"],
"name" : "Sum_650",
"op" : "Sum",
"outputs" : ["Sum_650_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_654"],
"name" : "Sum_655",
"op" : "Sum",
"outputs" : ["Sum_655_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_659"],
"name" : "Sum_660",
"op" : "Sum",
"outputs" : ["Sum_660_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_664"],
"name" : "Sum_665",
"op" : "Sum",
"outputs" : ["Sum_665_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_669"],
"name" : "Sum_670",
"op" : "Sum",
"outputs" : ["Sum_670_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_674"],
"name" : "Sum_675",
"op" : "Sum",
"outputs" : ["Sum_675_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_679"],
"name" : "Sum_680",
"op" : "Sum",
"outputs" : ["Sum_680_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_684"],
"name" : "Sum_685",
"op" : "Sum",
"outputs" : ["Sum_685_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_689"],
"name" : "Sum_690",
"op" : "Sum",
"outputs" : ["Sum_690_0"],
"reduction_axes" : [ 0, 1, 2, 3 ]
},
{
"inputs" : ["Multiply_694"],
"name" : "Sum_695",
"op" : "Sum",
"outputs" : ["Sum_695_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_699"],
"name" : "Sum_700",
"op" : "Sum",
"outputs" : ["Sum_700_0"],
"reduction_axes" : [0]
},
{
"inputs" : ["Multiply_704"],
"name" : "Sum_705",
"op" : "Sum",
"outputs" : ["Sum_705_0"],
"reduction_axes" : [ 0, 1 ]
},
{
"inputs" : ["Multiply_709"],
"name" : "Sum_710",
"op" : "Sum",
"outputs" : ["Sum_710_0"],
"reduction_axes" : [0]
},
{
"inputs" : [ "Broadcast_511", "Broadcast_510" ],
"name" : "Multiply_512",
"op" : "Multiply",
"outputs" : ["Multiply_512_0"]
},
{
"inputs" : [ "Broadcast_519", "Parameter_507" ],
"name" : "Maximum_520",
"op" : "Maximum",
"outputs" : ["Maximum_520_0"]
},
{
"inputs" : [ "Sum_586", "Constant_587" ],
"name" : "Divide_588",
"op" : "Divide",
"outputs" : ["Divide_588_0"]
},
{
"inputs" : [ "Sum_590", "Constant_591" ],
"name" : "Divide_592",
"op" : "Divide",
"outputs" : ["Divide_592_0"]
},
{
"inputs" : [ "Sum_595", "Constant_596" ],
"name" : "Divide_597",
"op" : "Divide",
"outputs" : ["Divide_597_0"]
},
{
"inputs" : [ "Sum_600", "Constant_601" ],
"name" : "Divide_602",
"op" : "Divide",
"outputs" : ["Divide_602_0"]
},
{
"inputs" : [ "Sum_605", "Constant_606" ],
"name" : "Divide_607",
"op" : "Divide",
"outputs" : ["Divide_607_0"]
},
{
"inputs" : [ "Sum_610", "Constant_611" ],
"name" : "Divide_612",
"op" : "Divide",
"outputs" : ["Divide_612_0"]
},
{
"inputs" : [ "Sum_615", "Constant_616" ],
"name" : "Divide_617",
"op" : "Divide",
"outputs" : ["Divide_617_0"]
},
{
"inputs" : [ "Sum_620", "Constant_621" ],
"name" : "Divide_622",
"op" : "Divide",
"outputs" : ["Divide_622_0"]
},
{
"inputs" : [ "Sum_625", "Constant_626" ],
"name" : "Divide_627",
"op" : "Divide",
"outputs" : ["Divide_627_0"]
},
{
"inputs" : [ "Sum_630", "Constant_631" ],
"name" : "Divide_632",
"op" : "Divide",
"outputs" : ["Divide_632_0"]
},
{
"inputs" : [ "Sum_635", "Constant_636" ],
"name" : "Divide_637",
"op" : "Divide",
"outputs" : ["Divide_637_0"]
},
{
"inputs" : [ "Sum_640", "Constant_641" ],
"name" : "Divide_642",
"op" : "Divide",
"outputs" : ["Divide_642_0"]
},
{
"inputs" : [ "Sum_645", "Constant_646" ],
"name" : "Divide_647",
"op" : "Divide",
"outputs" : ["Divide_647_0"]
},
{
"inputs" : [ "Sum_650", "Constant_651" ],
"name" : "Divide_652",
"op" : "Divide",
"outputs" : ["Divide_652_0"]
},
{
"inputs" : [ "Sum_655", "Constant_656" ],
"name" : "Divide_657",
"op" : "Divide",
"outputs" : ["Divide_657_0"]
},
{
"inputs" : [ "Sum_660", "Constant_661" ],
"name" : "Divide_662",
"op" : "Divide",
"outputs" : ["Divide_662_0"]
},
{
"inputs" : [ "Sum_665", "Constant_666" ],
"name" : "Divide_667",
"op" : "Divide",
"outputs" : ["Divide_667_0"]
},
{
"inputs" : [ "Sum_670", "Constant_671" ],
"name" : "Divide_672",
"op" : "Divide",
"outputs" : ["Divide_672_0"]
},
{
"inputs" : [ "Sum_675", "Constant_676" ],
"name" : "Divide_677",
"op" : "Divide",
"outputs" : ["Divide_677_0"]
},
{
"inputs" : [ "Sum_680", "Constant_681" ],
"name" : "Divide_682",
"op" : "Divide",
"outputs" : ["Divide_682_0"]
},
{
"inputs" : [ "Sum_685", "Constant_686" ],
"name" : "Divide_687",
"op" : "Divide",
"outputs" : ["Divide_687_0"]
},
{
"inputs" : [ "Sum_690", "Constant_691" ],
"name" : "Divide_692",
"op" : "Divide",
"outputs" : ["Divide_692_0"]
},
{
"inputs" : [ "Sum_695", "Constant_696" ],
"name" : "Divide_697",
"op" : "Divide",
"outputs" : ["Divide_697_0"]
},
{
"inputs" : [ "Sum_700", "Constant_701" ],
"name" : "Divide_702",
"op" : "Divide",
"outputs" : ["Divide_702_0"]
},
{
"inputs" : [ "Sum_705", "Constant_706" ],
"name" : "Divide_707",
"op" : "Divide",
"outputs" : ["Divide_707_0"]
},
{
"inputs" : [ "Sum_710", "Constant_711" ],
"name" : "Divide_712",
"op" : "Divide",
"outputs" : ["Divide_712_0"]
},
{
"inputs" : ["Multiply_512"],
"name" : "Sum_513",
"op" : "Sum",
"outputs" : ["Sum_513_0"],
"reduction_axes" : [0]
},
{
"inputs" : [ "Maximum_520", "Broadcast_575" ],
"name" : "Greater_576",
"op" : "Greater",
"outputs" : ["Greater_576_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Maximum_520"],
"name" : "Reshape_521",
"op" : "Reshape",
"output_shape" : [ 50, 64, 8, 8 ],
"outputs" : ["Reshape_521_0"]
},
{
"inputs" : [ "Divide_588", "Divide_592" ],
"name" : "Add_593",
"op" : "Add",
"outputs" : ["Add_593_0"]
},
{
"inputs" : [ "Sum_513", "Constant_514" ],
"name" : "Equal_544",
"op" : "Equal",
"outputs" : ["Equal_544_0"]
},
{
"inputs" : [ "Sum_513", "Constant_514" ],
"name" : "Greater_515",
"op" : "Greater",
"outputs" : ["Greater_515_0"]
},
{
"include_padding_in_avg_computation" : false,
"inputs" : ["Reshape_521"],
"name" : "AvgPool_522",
"op" : "AvgPool",
"outputs" : ["AvgPool_522_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_movement_strides" : [ 1, 1 ],
"window_shape" : [ 8, 8 ]
},
{
"inputs" : [ "Add_593", "Divide_597" ],
"name" : "Add_598",
"op" : "Add",
"outputs" : ["Add_598_0"]
},
{
"inputs" : [ "Equal_544", "Constant_508", "Sum_513" ],
"name" : "Select_545",
"op" : "Select",
"outputs" : ["Select_545_0"]
},
{
"inputs" : [ "Greater_515", "Constant_508", "Constant_550" ],
"name" : "Select_551",
"op" : "Select",
"outputs" : ["Select_551_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["AvgPool_522"],
"name" : "Reshape_523",
"op" : "Reshape",
"output_shape" : [ 50, 1, 1, 64 ],
"outputs" : ["Reshape_523_0"]
},
{
"inputs" : [ "Add_598", "Divide_602" ],
"name" : "Add_603",
"op" : "Add",
"outputs" : ["Add_603_0"]
},
{
"inputs" : [ "Select_551", "Select_545" ],
"name" : "Divide_552",
"op" : "Divide",
"outputs" : ["Divide_552_0"]
},
{
"input_order" : [ 0, 1, 2, 3 ],
"inputs" : ["Reshape_523"],
"name" : "Reshape_524",
"op" : "Reshape",
"output_shape" : [ 50, 64 ],
"outputs" : ["Reshape_524_0"]
},
{
"inputs" : [ "Add_603", "Divide_607" ],
"name" : "Add_608",
"op" : "Add",
"outputs" : ["Add_608_0"]
},
{
"input_order" : [],
"inputs" : ["Divide_552"],
"name" : "Reshape_553",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_553_0"]
},
{
"inputs" : [ "Reshape_524", "Parameter_481" ],
"name" : "Dot_525",
"op" : "Dot",
"outputs" : ["Dot_525_0"],
"reduction_axes_count" : 1
},
{
"input_order" : [ 1, 0 ],
"inputs" : ["Reshape_524"],
"name" : "Reshape_572",
"op" : "Reshape",
"output_shape" : [ 64, 50 ],
"outputs" : ["Reshape_572_0"]
},
{
"inputs" : [ "Add_608", "Divide_612" ],
"name" : "Add_613",
"op" : "Add",
"outputs" : ["Add_613_0"]
},
{
"input_order" : [],
"inputs" : ["Reshape_553"],
"name" : "Reshape_554",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_554_0"]
},
{
"inputs" : [ "Dot_525", "Broadcast_526" ],
"name" : "Add_527",
"op" : "Add",
"outputs" : ["Add_527_0"]
},
{
"inputs" : [ "Add_613", "Divide_617" ],
"name" : "Add_618",
"op" : "Add",
"outputs" : ["Add_618_0"]
},
{
"input_order" : [],
"inputs" : ["Reshape_554"],
"name" : "Reshape_555",
"op" : "Reshape",
"output_shape" : [1],
"outputs" : ["Reshape_555_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Add_527"],
"name" : "Reshape_528",
"op" : "Reshape",
"output_shape" : [ 50, 10 ],
"outputs" : ["Reshape_528_0"]
},
{
"inputs" : [ "Add_618", "Divide_622" ],
"name" : "Add_623",
"op" : "Add",
"outputs" : ["Add_623_0"]
},
{
"input_order" : [0],
"inputs" : ["Reshape_555"],
"name" : "Reshape_556",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_556_0"]
},
{
"inputs" : ["Reshape_528"],
"name" : "Max_529",
"op" : "Max",
"outputs" : ["Max_529_0"],
"reduction_axes" : [1]
},
{
"inputs" : [ "Add_623", "Divide_627" ],
"name" : "Add_628",
"op" : "Add",
"outputs" : ["Add_628_0"]
},
{
"axes" : [0],
"inputs" : ["Reshape_556"],
"name" : "Broadcast_557",
"op" : "Broadcast",
"outputs" : ["Broadcast_557_0"],
"shape" : [50]
},
{
"axes" : [1],
"inputs" : ["Max_529"],
"name" : "Broadcast_530",
"op" : "Broadcast",
"outputs" : ["Broadcast_530_0"],
"shape" : [ 50, 10 ]
},
{
"inputs" : [ "Add_628", "Divide_632" ],
"name" : "Add_633",
"op" : "Add",
"outputs" : ["Add_633_0"]
},
{
"inputs" : [ "Broadcast_549", "Broadcast_557" ],
"name" : "Add_558",
"op" : "Add",
"outputs" : ["Add_558_0"]
},
{
"inputs" : [ "Reshape_528", "Broadcast_530" ],
"name" : "Subtract_531",
"op" : "Subtract",
"outputs" : ["Subtract_531_0"]
},
{
"inputs" : [ "Add_633", "Divide_637" ],
"name" : "Add_638",
"op" : "Add",
"outputs" : ["Add_638_0"]
},
{
"inputs" : [ "Add_558", "Broadcast_559" ],
"name" : "Multiply_560",
"op" : "Multiply",
"outputs" : ["Multiply_560_0"]
},
{
"inputs" : ["Subtract_531"],
"name" : "Exp_532",
"op" : "Exp",
"outputs" : ["Exp_532_0"]
},
{
"inputs" : [ "Add_638", "Divide_642" ],
"name" : "Add_643",
"op" : "Add",
"outputs" : ["Add_643_0"]
},
{
"input_order" : [0],
"inputs" : ["Multiply_560"],
"name" : "Reshape_561",
"op" : "Reshape",
"output_shape" : [50],
"outputs" : ["Reshape_561_0"]
},
{
"inputs" : ["Exp_532"],
"name" : "Sum_533",
"op" : "Sum",
"outputs" : ["Sum_533_0"],
"reduction_axes" : [1]
},
{
"inputs" : [ "Add_643", "Divide_647" ],
"name" : "Add_648",
"op" : "Add",
"outputs" : ["Add_648_0"]
},
{
"input_order" : [0],
"inputs" : ["Reshape_561"],
"name" : "Reshape_562",
"op" : "Reshape",
"output_shape" : [50],
"outputs" : ["Reshape_562_0"]
},
{
"axes" : [1],
"inputs" : ["Sum_533"],
"name" : "Broadcast_564",
"op" : "Broadcast",
"outputs" : ["Broadcast_564_0"],
"shape" : [ 50, 10 ]
},
{
"inputs" : ["Sum_533"],
"name" : "Log_534",
"op" : "Log",
"outputs" : ["Log_534_0"]
},
{
"inputs" : [ "Add_648", "Divide_652" ],
"name" : "Add_653",
"op" : "Add",
"outputs" : ["Add_653_0"]
},
{
"input_order" : [0],
"inputs" : ["Reshape_562"],
"name" : "Reshape_563",
"op" : "Reshape",
"output_shape" : [ 50, 1 ],
"outputs" : ["Reshape_563_0"]
},
{
"inputs" : [ "Exp_532", "Broadcast_564" ],
"name" : "Divide_565",
"op" : "Divide",
"outputs" : ["Divide_565_0"]
},
{
"axes" : [1],
"inputs" : ["Log_534"],
"name" : "Broadcast_535",
"op" : "Broadcast",
"outputs" : ["Broadcast_535_0"],
"shape" : [ 50, 10 ]
},
{
"inputs" : [ "Add_653", "Divide_657" ],
"name" : "Add_658",
"op" : "Add",
"outputs" : ["Add_658_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Reshape_563"],
"name" : "Reshape_567",
"op" : "Reshape",
"output_shape" : [50],
"outputs" : ["Reshape_567_0"]
},
{
"inputs" : [ "Divide_565", "Reshape_516" ],
"name" : "Subtract_566",
"op" : "Subtract",
"outputs" : ["Subtract_566_0"]
},
{
"inputs" : [ "Subtract_531", "Broadcast_535" ],
"name" : "Subtract_536",
"op" : "Subtract",
"outputs" : ["Subtract_536_0"]
},
{
"inputs" : [ "Add_658", "Divide_662" ],
"name" : "Add_663",
"op" : "Add",
"outputs" : ["Add_663_0"]
},
{
"axes" : [1],
"inputs" : ["Reshape_567"],
"name" : "Broadcast_568",
"op" : "Broadcast",
"outputs" : ["Broadcast_568_0"],
"shape" : [ 50, 10 ]
},
{
"inputs" : [ "Negative_517", "Subtract_536" ],
"name" : "Multiply_537",
"op" : "Multiply",
"outputs" : ["Multiply_537_0"]
},
{
"inputs" : [ "Add_663", "Divide_667" ],
"name" : "Add_668",
"op" : "Add",
"outputs" : ["Add_668_0"]
},
{
"inputs" : [ "Broadcast_568", "Subtract_566" ],
"name" : "Multiply_569",
"op" : "Multiply",
"outputs" : ["Multiply_569_0"]
},
{
"inputs" : ["Multiply_537"],
"name" : "Sum_538",
"op" : "Sum",
"outputs" : ["Sum_538_0"],
"reduction_axes" : [1]
},
{
"inputs" : [ "Add_668", "Divide_672" ],
"name" : "Add_673",
"op" : "Add",
"outputs" : ["Add_673_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Multiply_569"],
"name" : "Reshape_570",
"op" : "Reshape",
"output_shape" : [ 50, 10 ],
"outputs" : ["Reshape_570_0"]
},
{
"input_order" : [0],
"inputs" : ["Sum_538"],
"name" : "Reshape_539",
"op" : "Reshape",
"output_shape" : [50],
"outputs" : ["Reshape_539_0"]
},
{
"inputs" : [ "Add_673", "Divide_677" ],
"name" : "Add_678",
"op" : "Add",
"outputs" : ["Add_678_0"]
},
{
"inputs" : [ "Reshape_570", "Reshape_577" ],
"name" : "Dot_578",
"op" : "Dot",
"outputs" : ["Dot_578_0"],
"reduction_axes_count" : 1
},
{
"inputs" : [ "Reshape_572", "Reshape_570" ],
"name" : "Dot_573",
"op" : "Dot",
"outputs" : ["Dot_573_0"],
"reduction_axes_count" : 1
},
{
"inputs" : ["Reshape_570"],
"name" : "Sum_571",
"op" : "Sum",
"outputs" : ["Sum_571_0"],
"reduction_axes" : [0]
},
{
"inputs" : [ "Reshape_539", "Broadcast_541" ],
"name" : "Multiply_542",
"op" : "Multiply",
"outputs" : ["Multiply_542_0"]
},
{
"inputs" : [ "Add_678", "Divide_682" ],
"name" : "Add_683",
"op" : "Add",
"outputs" : ["Add_683_0"]
},
{
"input_order" : [ 0, 1 ],
"inputs" : ["Dot_578"],
"name" : "Reshape_579",
"op" : "Reshape",
"output_shape" : [ 50, 1, 1, 64 ],
"outputs" : ["Reshape_579_0"]
},
{
"inputs" : ["Multiply_542"],
"name" : "Sum_543",
"op" : "Sum",
"outputs" : ["Sum_543_0"],
"reduction_axes" : [0]
},
{
"inputs" : [ "Add_683", "Divide_687" ],
"name" : "Add_688",
"op" : "Add",
"outputs" : ["Add_688_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Reshape_579"],
"name" : "Reshape_580",
"op" : "Reshape",
"output_shape" : [ 50, 64, 1, 1 ],
"outputs" : ["Reshape_580_0"]
},
{
"inputs" : [ "Sum_543", "Select_545" ],
"name" : "Divide_546",
"op" : "Divide",
"outputs" : ["Divide_546_0"]
},
{
"inputs" : [ "Add_688", "Divide_692" ],
"name" : "Add_693",
"op" : "Add",
"outputs" : ["Add_693_0"]
},
{
"forward_arg_shape" : [ 50, 64, 8, 8 ],
"include_padding_in_avg_computation" : false,
"inputs" : ["Reshape_580"],
"name" : "AvgPoolBackprop_581",
"op" : "AvgPoolBackprop",
"outputs" : ["AvgPoolBackprop_581_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_movement_strides" : [ 1, 1 ],
"window_shape" : [ 8, 8 ]
},
{
"inputs" : [ "Greater_515", "Divide_546", "Constant_514" ],
"name" : "Select_547",
"op" : "Select",
"outputs" : ["Select_547_0"]
},
{
"inputs" : [ "Add_693", "Divide_697" ],
"name" : "Add_698",
"op" : "Add",
"outputs" : ["Add_698_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["AvgPoolBackprop_581"],
"name" : "Reshape_582",
"op" : "Reshape",
"output_shape" : [ 50, 8, 8, 64 ],
"outputs" : ["Reshape_582_0"]
},
{
"inputs" : [ "Add_698", "Divide_702" ],
"name" : "Add_703",
"op" : "Add",
"outputs" : ["Add_703_0"]
},
{
"inputs" : [ "Greater_576", "Reshape_582", "Broadcast_575" ],
"name" : "Select_583",
"op" : "Select",
"outputs" : ["Select_583_0"]
},
{
"inputs" : [ "Add_703", "Divide_707" ],
"name" : "Add_708",
"op" : "Add",
"outputs" : ["Add_708_0"]
},
{
"inputs" : [ "Add_708", "Divide_712" ],
"name" : "Add_713",
"op" : "Add",
"outputs" : ["Add_713_0"]
},
{
"inputs" : [ "Constant_584", "Add_713" ],
"name" : "Multiply_714",
"op" : "Multiply",
"outputs" : ["Multiply_714_0"]
},
{
"inputs" : [ "Select_547", "Multiply_714" ],
"name" : "Add_715",
"op" : "Add",
"outputs" : ["Add_715_0"]
}
],
"parameters" : [
"Parameter_480", "Parameter_481", "Parameter_482", "Parameter_483",
"Parameter_484", "Parameter_485", "Parameter_486", "Parameter_487",
"Parameter_488", "Parameter_489", "Parameter_490", "Parameter_491",
"Parameter_492", "Parameter_493", "Parameter_494", "Parameter_495",
"Parameter_496", "Parameter_497", "Parameter_498", "Parameter_499",
"Parameter_500", "Parameter_501", "Parameter_502", "Parameter_503",
"Parameter_504", "Parameter_505", "Parameter_506", "Parameter_507"
],
"result" : [
"Select_547", "Add_527", "Sum_571", "Dot_573", "Select_583", "Add_715",
"Select_547"
]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_28[_XlaCompiledKernel=true,_XlaNumConstantArgs=0,_XlaNumResourceArgs=0].v13.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_7",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_412",
"op" : "Parameter",
"outputs" : ["Parameter_412_0"],
"shape" : [ 1, 1, 32, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_411",
"op" : "Parameter",
"outputs" : ["Parameter_411_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_410",
"op" : "Parameter",
"outputs" : ["Parameter_410_0"],
"shape" : [ 3, 3, 64, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_409",
"op" : "Parameter",
"outputs" : ["Parameter_409_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_413",
"op" : "Constant",
"outputs" : ["Constant_413_0"],
"shape" : [],
"value" : ["0"]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_420",
"op" : "Constant",
"outputs" : ["Constant_420_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_412"],
"name" : "Reshape_423",
"op" : "Reshape",
"output_shape" : [ 64, 32, 1, 1 ],
"outputs" : ["Reshape_423_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_410"],
"name" : "Reshape_417",
"op" : "Reshape",
"output_shape" : [ 64, 64, 3, 3 ],
"outputs" : ["Reshape_417_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_413"],
"name" : "Broadcast_414",
"op" : "Broadcast",
"outputs" : ["Broadcast_414_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"inputs" : [ "Parameter_411", "Constant_420" ],
"name" : "Pad_421",
"op" : "Pad",
"outputs" : ["Pad_421_0"],
"padding_above" : [ 0, 0, 0, 0 ],
"padding_below" : [ 0, 0, 0, 0 ],
"padding_interior" : [ 0, 0, 0, 0 ]
},
{
"inputs" : [ "Broadcast_414", "Parameter_409" ],
"name" : "Maximum_415",
"op" : "Maximum",
"outputs" : ["Maximum_415_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Pad_421"],
"name" : "Reshape_422",
"op" : "Reshape",
"output_shape" : [ 50, 32, 16, 16 ],
"outputs" : ["Reshape_422_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Maximum_415"],
"name" : "Reshape_416",
"op" : "Reshape",
"output_shape" : [ 50, 64, 8, 8 ],
"outputs" : ["Reshape_416_0"]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_422", "Reshape_423" ],
"name" : "Convolution_424",
"op" : "Convolution",
"outputs" : ["Convolution_424_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 2, 2 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_416", "Reshape_417" ],
"name" : "Convolution_418",
"op" : "Convolution",
"outputs" : ["Convolution_418_0"],
"padding_above" : [ 1, 1 ],
"padding_below" : [ 1, 1 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["Convolution_424"],
"name" : "Reshape_425",
"op" : "Reshape",
"output_shape" : [ 50, 8, 8, 64 ],
"outputs" : ["Reshape_425_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["Convolution_418"],
"name" : "Reshape_419",
"op" : "Reshape",
"output_shape" : [ 50, 8, 8, 64 ],
"outputs" : ["Reshape_419_0"]
},
{
"inputs" : [ "Reshape_419", "Reshape_425" ],
"name" : "Add_426",
"op" : "Add",
"outputs" : ["Add_426_0"]
}
],
"parameters" :
[ "Parameter_409", "Parameter_410", "Parameter_411", "Parameter_412" ],
"result" :
[ "Add_426", "Pad_421", "Maximum_415", "Reshape_425", "Reshape_419" ]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_34[_XlaCompiledKernel=true,_XlaNumConstantArgs=7,_XlaNumResourceArgs=0].v73.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_10",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_849",
"op" : "Parameter",
"outputs" : ["Parameter_849_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_848",
"op" : "Parameter",
"outputs" : ["Parameter_848_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_847",
"op" : "Parameter",
"outputs" : ["Parameter_847_0"],
"shape" : [ 50, 18, 18, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_846",
"op" : "Parameter",
"outputs" : ["Parameter_846_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_845",
"op" : "Parameter",
"outputs" : ["Parameter_845_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_844",
"op" : "Parameter",
"outputs" : ["Parameter_844_0"],
"shape" : [ 3, 3, 64, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_843",
"op" : "Parameter",
"outputs" : ["Parameter_843_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_842",
"op" : "Parameter",
"outputs" : ["Parameter_842_0"],
"shape" : [ 50, 8, 8, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_841",
"op" : "Parameter",
"outputs" : ["Parameter_841_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_840",
"op" : "Parameter",
"outputs" : ["Parameter_840_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_839",
"op" : "Parameter",
"outputs" : ["Parameter_839_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_838",
"op" : "Parameter",
"outputs" : ["Parameter_838_0"],
"shape" : [ 64, 10 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_837",
"op" : "Parameter",
"outputs" : ["Parameter_837_0"],
"shape" : [10]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_836",
"op" : "Parameter",
"outputs" : ["Parameter_836_0"],
"shape" : [ 1, 1, 16, 32 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_835",
"op" : "Parameter",
"outputs" : ["Parameter_835_0"],
"shape" : [ 1, 1, 32, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_834",
"op" : "Parameter",
"outputs" : ["Parameter_834_0"],
"shape" : [ 3, 3, 32, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_833",
"op" : "Parameter",
"outputs" : ["Parameter_833_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_832",
"op" : "Parameter",
"outputs" : ["Parameter_832_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_831",
"op" : "Parameter",
"outputs" : ["Parameter_831_0"],
"shape" : [ 3, 3, 64, 64 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_830",
"op" : "Parameter",
"outputs" : ["Parameter_830_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_829",
"op" : "Parameter",
"outputs" : ["Parameter_829_0"],
"shape" : [64]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_828",
"op" : "Parameter",
"outputs" : ["Parameter_828_0"],
"shape" : [ 64, 10 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_827",
"op" : "Parameter",
"outputs" : ["Parameter_827_0"],
"shape" : [10]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_826",
"op" : "Parameter",
"outputs" : ["Parameter_826_0"],
"shape" : []
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_851",
"op" : "Constant",
"outputs" : ["Constant_851_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Parameter_848"],
"name" : "Reshape_854",
"op" : "Reshape",
"output_shape" : [ 50, 64, 8, 8 ],
"outputs" : ["Reshape_854_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_848"],
"name" : "Reshape_888",
"op" : "Reshape",
"output_shape" : [ 64, 50, 8, 8 ],
"outputs" : ["Reshape_888_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_847"],
"name" : "Reshape_887",
"op" : "Reshape",
"output_shape" : [ 32, 50, 18, 18 ],
"outputs" : ["Reshape_887_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_842"],
"name" : "Reshape_897",
"op" : "Reshape",
"output_shape" : [ 64, 50, 8, 8 ],
"outputs" : ["Reshape_897_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_841"],
"name" : "Reshape_896",
"op" : "Reshape",
"output_shape" : [ 32, 50, 16, 16 ],
"outputs" : ["Reshape_896_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_834"],
"name" : "Reshape_855",
"op" : "Reshape",
"output_shape" : [ 64, 32, 3, 3 ],
"outputs" : ["Reshape_855_0"]
},
{
"input_order" : [],
"inputs" : ["Parameter_826"],
"name" : "Reshape_850",
"op" : "Reshape",
"output_shape" : [],
"outputs" : ["Reshape_850_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_851"],
"name" : "Broadcast_852",
"op" : "Broadcast",
"outputs" : ["Broadcast_852_0"],
"shape" : [ 50, 16, 16, 32 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_887", "Reshape_888" ],
"name" : "Convolution_889",
"op" : "Convolution",
"outputs" : ["Convolution_889_0"],
"padding_above" : [ -1, -1 ],
"padding_below" : [ 0, 0 ],
"window_dilation_strides" : [ 2, 2 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_896", "Reshape_897" ],
"name" : "Convolution_898",
"op" : "Convolution",
"outputs" : ["Convolution_898_0"],
"padding_above" : [ -1, -1 ],
"padding_below" : [ 0, 0 ],
"window_dilation_strides" : [ 2, 2 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_batch_shape" : [ 50, 32, 18, 18 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_855", "Reshape_854" ],
"name" : "ConvolutionBackpropData_856",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_856_0"],
"padding_above_forward" : [ 0, 0 ],
"padding_below_forward" : [ 0, 0 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 2, 2 ]
},
{
"axes" : [0],
"inputs" : ["Reshape_850"],
"name" : "Broadcast_873",
"op" : "Broadcast",
"outputs" : ["Broadcast_873_0"],
"shape" : [64]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Reshape_850"],
"name" : "Broadcast_876",
"op" : "Broadcast",
"outputs" : ["Broadcast_876_0"],
"shape" : [ 3, 3, 64, 64 ]
},
{
"axes" : [ 0, 1 ],
"inputs" : ["Reshape_850"],
"name" : "Broadcast_867",
"op" : "Broadcast",
"outputs" : ["Broadcast_867_0"],
"shape" : [ 64, 10 ]
},
{
"axes" : [0],
"inputs" : ["Reshape_850"],
"name" : "Broadcast_879",
"op" : "Broadcast",
"outputs" : ["Broadcast_879_0"],
"shape" : [64]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Reshape_850"],
"name" : "Broadcast_885",
"op" : "Broadcast",
"outputs" : ["Broadcast_885_0"],
"shape" : [ 3, 3, 32, 64 ]
},
{
"input_order" : [],
"inputs" : ["Reshape_850"],
"name" : "Reshape_893",
"op" : "Reshape",
"output_shape" : [ 1, 1 ],
"outputs" : ["Reshape_893_0"]
},
{
"axes" : [0],
"inputs" : ["Reshape_850"],
"name" : "Broadcast_864",
"op" : "Broadcast",
"outputs" : ["Broadcast_864_0"],
"shape" : [10]
},
{
"axes" : [0],
"inputs" : ["Reshape_850"],
"name" : "Broadcast_882",
"op" : "Broadcast",
"outputs" : ["Broadcast_882_0"],
"shape" : [64]
},
{
"axes" : [0],
"inputs" : ["Reshape_850"],
"name" : "Broadcast_870",
"op" : "Broadcast",
"outputs" : ["Broadcast_870_0"],
"shape" : [64]
},
{
"input_order" : [],
"inputs" : ["Reshape_850"],
"name" : "Reshape_861",
"op" : "Reshape",
"output_shape" : [ 1, 1 ],
"outputs" : ["Reshape_861_0"]
},
{
"inputs" : [ "Parameter_849", "Broadcast_852" ],
"name" : "Greater_853",
"op" : "Greater",
"outputs" : ["Greater_853_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_889"],
"name" : "Reshape_890",
"op" : "Reshape",
"output_shape" : [ 64, 3, 3, 32 ],
"outputs" : ["Reshape_890_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_898"],
"name" : "Reshape_899",
"op" : "Reshape",
"output_shape" : [ 64, 1, 1, 32 ],
"outputs" : ["Reshape_899_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_856"],
"name" : "Reshape_857",
"op" : "Reshape",
"output_shape" : [ 50, 18, 18, 32 ],
"outputs" : ["Reshape_857_0"]
},
{
"inputs" : [ "Parameter_830", "Broadcast_873" ],
"name" : "Multiply_874",
"op" : "Multiply",
"outputs" : ["Multiply_874_0"]
},
{
"inputs" : [ "Parameter_831", "Broadcast_876" ],
"name" : "Multiply_877",
"op" : "Multiply",
"outputs" : ["Multiply_877_0"]
},
{
"inputs" : [ "Parameter_828", "Broadcast_867" ],
"name" : "Multiply_868",
"op" : "Multiply",
"outputs" : ["Multiply_868_0"]
},
{
"inputs" : [ "Parameter_832", "Broadcast_879" ],
"name" : "Multiply_880",
"op" : "Multiply",
"outputs" : ["Multiply_880_0"]
},
{
"inputs" : [ "Parameter_834", "Broadcast_885" ],
"name" : "Multiply_886",
"op" : "Multiply",
"outputs" : ["Multiply_886_0"]
},
{
"axes" : [ 2, 3 ],
"inputs" : ["Reshape_893"],
"name" : "Broadcast_894",
"op" : "Broadcast",
"outputs" : ["Broadcast_894_0"],
"shape" : [ 1, 1, 32, 64 ]
},
{
"inputs" : [ "Parameter_827", "Broadcast_864" ],
"name" : "Multiply_865",
"op" : "Multiply",
"outputs" : ["Multiply_865_0"]
},
{
"inputs" : [ "Parameter_833", "Broadcast_882" ],
"name" : "Multiply_883",
"op" : "Multiply",
"outputs" : ["Multiply_883_0"]
},
{
"inputs" : [ "Parameter_829", "Broadcast_870" ],
"name" : "Multiply_871",
"op" : "Multiply",
"outputs" : ["Multiply_871_0"]
},
{
"axes" : [ 2, 3 ],
"inputs" : ["Reshape_861"],
"name" : "Broadcast_862",
"op" : "Broadcast",
"outputs" : ["Broadcast_862_0"],
"shape" : [ 1, 1, 16, 32 ]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_890"],
"name" : "Reshape_891",
"op" : "Reshape",
"output_shape" : [ 3, 3, 32, 64 ],
"outputs" : ["Reshape_891_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_899"],
"name" : "Reshape_900",
"op" : "Reshape",
"output_shape" : [ 1, 1, 32, 64 ],
"outputs" : ["Reshape_900_0"]
},
{
"inputs" : ["Reshape_857"],
"lower_bounds" : [ 0, 1, 1, 0 ],
"name" : "Slice_858",
"op" : "Slice",
"outputs" : ["Slice_858_0"],
"strides" : [ 1, 1, 1, 1 ],
"upper_bounds" : [ 50, 17, 17, 32 ]
},
{
"inputs" : [ "Multiply_874", "Parameter_839" ],
"name" : "Add_875",
"op" : "Add",
"outputs" : ["Add_875_0"]
},
{
"inputs" : [ "Multiply_877", "Parameter_844" ],
"name" : "Add_878",
"op" : "Add",
"outputs" : ["Add_878_0"]
},
{
"inputs" : [ "Multiply_868", "Parameter_838" ],
"name" : "Add_869",
"op" : "Add",
"outputs" : ["Add_869_0"]
},
{
"inputs" : [ "Multiply_880", "Parameter_846" ],
"name" : "Add_881",
"op" : "Add",
"outputs" : ["Add_881_0"]
},
{
"inputs" : [ "Parameter_835", "Broadcast_894" ],
"name" : "Multiply_895",
"op" : "Multiply",
"outputs" : ["Multiply_895_0"]
},
{
"inputs" : [ "Multiply_865", "Parameter_837" ],
"name" : "Add_866",
"op" : "Add",
"outputs" : ["Add_866_0"]
},
{
"inputs" : [ "Multiply_883", "Parameter_845" ],
"name" : "Add_884",
"op" : "Add",
"outputs" : ["Add_884_0"]
},
{
"inputs" : [ "Multiply_871", "Parameter_840" ],
"name" : "Add_872",
"op" : "Add",
"outputs" : ["Add_872_0"]
},
{
"inputs" : [ "Parameter_836", "Broadcast_862" ],
"name" : "Multiply_863",
"op" : "Multiply",
"outputs" : ["Multiply_863_0"]
},
{
"inputs" : [ "Multiply_886", "Reshape_891" ],
"name" : "Add_892",
"op" : "Add",
"outputs" : ["Add_892_0"]
},
{
"inputs" : [ "Parameter_843", "Slice_858" ],
"name" : "Add_859",
"op" : "Add",
"outputs" : ["Add_859_0"]
},
{
"inputs" : [ "Multiply_895", "Reshape_900" ],
"name" : "Add_901",
"op" : "Add",
"outputs" : ["Add_901_0"]
},
{
"inputs" : [ "Greater_853", "Add_859", "Broadcast_852" ],
"name" : "Select_860",
"op" : "Select",
"outputs" : ["Select_860_0"]
}
],
"parameters" : [
"Parameter_826", "Parameter_827", "Parameter_828", "Parameter_829",
"Parameter_830", "Parameter_831", "Parameter_832", "Parameter_833",
"Parameter_834", "Parameter_835", "Parameter_836", "Parameter_837",
"Parameter_838", "Parameter_839", "Parameter_840", "Parameter_841",
"Parameter_842", "Parameter_843", "Parameter_844", "Parameter_845",
"Parameter_846", "Parameter_847", "Parameter_848", "Parameter_849"
],
"result" : [
"Reshape_850", "Select_860", "Multiply_863", "Add_866", "Add_869",
"Add_872", "Add_875", "Add_878", "Add_881", "Add_884", "Add_892",
"Add_901"
]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_4[_XlaCompiledKernel=true,_XlaNumConstantArgs=1,_XlaNumResourceArgs=0].v14.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_25",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1713",
"op" : "Parameter",
"outputs" : ["Parameter_1713_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1712",
"op" : "Parameter",
"outputs" : ["Parameter_1712_0"],
"shape" : [ 50, 32, 32, 3 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1711",
"op" : "Parameter",
"outputs" : ["Parameter_1711_0"],
"shape" : [ 3, 3, 3, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1710",
"op" : "Parameter",
"outputs" : ["Parameter_1710_0"],
"shape" : []
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1713"],
"name" : "Reshape_1717",
"op" : "Reshape",
"output_shape" : [ 16, 50, 32, 32 ],
"outputs" : ["Reshape_1717_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1712"],
"name" : "Reshape_1716",
"op" : "Reshape",
"output_shape" : [ 3, 50, 32, 32 ],
"outputs" : ["Reshape_1716_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1710"],
"name" : "Broadcast_1714",
"op" : "Broadcast",
"outputs" : ["Broadcast_1714_0"],
"shape" : [ 3, 3, 3, 16 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_1716", "Reshape_1717" ],
"name" : "Convolution_1718",
"op" : "Convolution",
"outputs" : ["Convolution_1718_0"],
"padding_above" : [ 1, 1 ],
"padding_below" : [ 1, 1 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"inputs" : [ "Parameter_1711", "Broadcast_1714" ],
"name" : "Multiply_1715",
"op" : "Multiply",
"outputs" : ["Multiply_1715_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_1718"],
"name" : "Reshape_1719",
"op" : "Reshape",
"output_shape" : [ 16, 3, 3, 3 ],
"outputs" : ["Reshape_1719_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_1719"],
"name" : "Reshape_1720",
"op" : "Reshape",
"output_shape" : [ 3, 3, 3, 16 ],
"outputs" : ["Reshape_1720_0"]
},
{
"inputs" : [ "Multiply_1715", "Reshape_1720" ],
"name" : "Add_1721",
"op" : "Add",
"outputs" : ["Add_1721_0"]
}
],
"parameters" : [
"Parameter_1710", "Parameter_1711", "Parameter_1712", "Parameter_1713"
],
"result" : ["Add_1721"]
}]
\ No newline at end of file
test/models/tensorflow/resnet8/tf_function_cluster_8[_XlaCompiledKernel=true,_XlaNumConstantArgs=2,_XlaNumResourceArgs=0].v28.json 0 → 100644
View file @ 7d34b79a
[{
"name" : "Function_22",
"ops" : [
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1540",
"op" : "Parameter",
"outputs" : ["Parameter_1540_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1539",
"op" : "Parameter",
"outputs" : ["Parameter_1539_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1538",
"op" : "Parameter",
"outputs" : ["Parameter_1538_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1537",
"op" : "Parameter",
"outputs" : ["Parameter_1537_0"],
"shape" : [ 1, 1, 16, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1536",
"op" : "Parameter",
"outputs" : ["Parameter_1536_0"],
"shape" : [ 3, 3, 16, 16 ]
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Parameter_1535",
"op" : "Parameter",
"outputs" : ["Parameter_1535_0"],
"shape" : []
},
{
"element_type" : "float",
"inputs" : [],
"name" : "Constant_1541",
"op" : "Constant",
"outputs" : ["Constant_1541_0"],
"shape" : [],
"value" : ["0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Parameter_1540"],
"name" : "Reshape_1548",
"op" : "Reshape",
"output_shape" : [ 50, 16, 32, 32 ],
"outputs" : ["Reshape_1548_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1540"],
"name" : "Reshape_1557",
"op" : "Reshape",
"output_shape" : [ 16, 50, 32, 32 ],
"outputs" : ["Reshape_1557_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1539"],
"name" : "Reshape_1565",
"op" : "Reshape",
"output_shape" : [ 16, 50, 32, 32 ],
"outputs" : ["Reshape_1565_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1539"],
"name" : "Reshape_1556",
"op" : "Reshape",
"output_shape" : [ 16, 50, 32, 32 ],
"outputs" : ["Reshape_1556_0"]
},
{
"input_order" : [ 3, 0, 1, 2 ],
"inputs" : ["Parameter_1538"],
"name" : "Reshape_1566",
"op" : "Reshape",
"output_shape" : [ 16, 50, 32, 32 ],
"outputs" : ["Reshape_1566_0"]
},
{
"input_order" : [ 0, 3, 1, 2 ],
"inputs" : ["Parameter_1538"],
"name" : "Reshape_1544",
"op" : "Reshape",
"output_shape" : [ 50, 16, 32, 32 ],
"outputs" : ["Reshape_1544_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_1537"],
"name" : "Reshape_1545",
"op" : "Reshape",
"output_shape" : [ 16, 16, 1, 1 ],
"outputs" : ["Reshape_1545_0"]
},
{
"input_order" : [ 3, 2, 0, 1 ],
"inputs" : ["Parameter_1536"],
"name" : "Reshape_1549",
"op" : "Reshape",
"output_shape" : [ 16, 16, 3, 3 ],
"outputs" : ["Reshape_1549_0"]
},
{
"input_order" : [],
"inputs" : ["Parameter_1535"],
"name" : "Reshape_1562",
"op" : "Reshape",
"output_shape" : [ 1, 1 ],
"outputs" : ["Reshape_1562_0"]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Parameter_1535"],
"name" : "Broadcast_1554",
"op" : "Broadcast",
"outputs" : ["Broadcast_1554_0"],
"shape" : [ 3, 3, 16, 16 ]
},
{
"axes" : [ 0, 1, 2, 3 ],
"inputs" : ["Constant_1541"],
"name" : "Broadcast_1542",
"op" : "Broadcast",
"outputs" : ["Broadcast_1542_0"],
"shape" : [ 50, 32, 32, 16 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_1556", "Reshape_1557" ],
"name" : "Convolution_1558",
"op" : "Convolution",
"outputs" : ["Convolution_1558_0"],
"padding_above" : [ 1, 1 ],
"padding_below" : [ 1, 1 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_dilation_strides" : [ 1, 1 ],
"inputs" : [ "Reshape_1565", "Reshape_1566" ],
"name" : "Convolution_1567",
"op" : "Convolution",
"outputs" : ["Convolution_1567_0"],
"padding_above" : [ 0, 0 ],
"padding_below" : [ 0, 0 ],
"window_dilation_strides" : [ 1, 1 ],
"window_movement_strides" : [ 1, 1 ]
},
{
"data_batch_shape" : [ 50, 16, 32, 32 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_1545", "Reshape_1544" ],
"name" : "ConvolutionBackpropData_1546",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_1546_0"],
"padding_above_forward" : [ 0, 0 ],
"padding_below_forward" : [ 0, 0 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 1, 1 ]
},
{
"data_batch_shape" : [ 50, 16, 32, 32 ],
"data_dilation_strides_forward" : [ 1, 1 ],
"inputs" : [ "Reshape_1549", "Reshape_1548" ],
"name" : "ConvolutionBackpropData_1550",
"op" : "ConvolutionBackpropData",
"outputs" : ["ConvolutionBackpropData_1550_0"],
"padding_above_forward" : [ 1, 1 ],
"padding_below_forward" : [ 1, 1 ],
"window_dilation_strides_forward" : [ 1, 1 ],
"window_movement_strides_forward" : [ 1, 1 ]
},
{
"axes" : [ 2, 3 ],
"inputs" : ["Reshape_1562"],
"name" : "Broadcast_1563",
"op" : "Broadcast",
"outputs" : ["Broadcast_1563_0"],
"shape" : [ 1, 1, 16, 16 ]
},
{
"inputs" : [ "Parameter_1536", "Broadcast_1554" ],
"name" : "Multiply_1555",
"op" : "Multiply",
"outputs" : ["Multiply_1555_0"]
},
{
"inputs" : [ "Parameter_1539", "Broadcast_1542" ],
"name" : "Greater_1543",
"op" : "Greater",
"outputs" : ["Greater_1543_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_1558"],
"name" : "Reshape_1559",
"op" : "Reshape",
"output_shape" : [ 16, 3, 3, 16 ],
"outputs" : ["Reshape_1559_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Convolution_1567"],
"name" : "Reshape_1568",
"op" : "Reshape",
"output_shape" : [ 16, 1, 1, 16 ],
"outputs" : ["Reshape_1568_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_1546"],
"name" : "Reshape_1547",
"op" : "Reshape",
"output_shape" : [ 50, 32, 32, 16 ],
"outputs" : ["Reshape_1547_0"]
},
{
"input_order" : [ 0, 2, 3, 1 ],
"inputs" : ["ConvolutionBackpropData_1550"],
"name" : "Reshape_1551",
"op" : "Reshape",
"output_shape" : [ 50, 32, 32, 16 ],
"outputs" : ["Reshape_1551_0"]
},
{
"inputs" : [ "Parameter_1537", "Broadcast_1563" ],
"name" : "Multiply_1564",
"op" : "Multiply",
"outputs" : ["Multiply_1564_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_1559"],
"name" : "Reshape_1560",
"op" : "Reshape",
"output_shape" : [ 3, 3, 16, 16 ],
"outputs" : ["Reshape_1560_0"]
},
{
"input_order" : [ 1, 2, 3, 0 ],
"inputs" : ["Reshape_1568"],
"name" : "Reshape_1569",
"op" : "Reshape",
"output_shape" : [ 1, 1, 16, 16 ],
"outputs" : ["Reshape_1569_0"]
},
{
"inputs" : [ "Reshape_1547", "Reshape_1551" ],
"name" : "Add_1552",
"op" : "Add",
"outputs" : ["Add_1552_0"]
},
{
"inputs" : [ "Multiply_1555", "Reshape_1560" ],
"name" : "Add_1561",
"op" : "Add",
"outputs" : ["Add_1561_0"]
},
{
"inputs" : [ "Multiply_1564", "Reshape_1569" ],
"name" : "Add_1570",
"op" : "Add",
"outputs" : ["Add_1570_0"]
},
{
"inputs" : [ "Greater_1543", "Add_1552", "Broadcast_1542" ],
"name" : "Select_1553",
"op" : "Select",
"outputs" : ["Select_1553_0"]
}
],
"parameters" : [
"Parameter_1535", "Parameter_1536", "Parameter_1537", "Parameter_1538",
"Parameter_1539", "Parameter_1540"
],
"result" : [ "Select_1553", "Add_1561", "Add_1570" ]
}]
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment