diff --git "a/model.json" "b/model.json" --- "a/model.json" +++ "b/model.json" @@ -1,3574 +1 @@ -{ - "format":"graph-model", - "generatedBy":"2.10.0-dev20220602", - "convertedBy":"TensorFlow.js Converter v3.18.0", - "signature":{ - "inputs":{ - "x":{ - "name":"x:0", - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"6" - }, - { - "size":"7" - } - ] - } - } - }, - "outputs":{ - "output_0":{ - "name":"Identity:0", - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"7" - } - ] - } - }, - "output_1":{ - "name":"Identity_1:0", - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - } - }, - "modelTopology":{ - "node":[ - { - "name":"StatefulPartitionedCall/transpose_69", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_38", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_22", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_91/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Cast_10", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_75", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"6" - }, - { - "size":"7" - }, - { - "size":"128" - }, - { - "size":"7" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_14", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"7" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/clip_by_value", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_INT32" - }, - "value":{ - "tensor":{ - "dtype":"DT_INT32", - "tensorShape":{ - "dim":[ - { - "size":"4" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/add", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_INT32" - }, - "value":{ - "tensor":{ - "dtype":"DT_INT32", - "tensorShape":{ - "dim":[ - { - "size":"4" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/ones_like", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_INT32", - "tensorShape":{ - "dim":[ - { - "size":"4" - } - ] - } - } - }, - "dtype":{ - "type":"DT_INT32" - } - } - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_12", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_10", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_8", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_6", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_4", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Reshape_1/shape", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_INT32" - }, - "value":{ - "tensor":{ - "dtype":"DT_INT32", - "tensorShape":{ - "dim":[ - { - "size":"4" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_3", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Mul", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_3/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_16/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_4", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_15/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Cast_1", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_15", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"3" - }, - { - "size":"3" - }, - { - "size":"128" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_32/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_8", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_31/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Cast_3", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_27", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"3" - }, - { - "size":"3" - }, - { - "size":"128" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_48/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_12", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_47/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Cast_5", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_39", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"3" - }, - { - "size":"3" - }, - { - "size":"128" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_64/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_16", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_63/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Cast_7", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_51", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"3" - }, - { - "size":"3" - }, - { - "size":"128" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_80/ReadVariableOp", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_20", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_79/ReadVariableOp", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Cast_9", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_63", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"3" - }, - { - "size":"3" - }, - { - "size":"128" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_81", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_42", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_24", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_101/ReadVariableOp", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Cast_11", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_87", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"6" - }, - { - "size":"7" - }, - { - "size":"128" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_45", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_26", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_109/ReadVariableOp", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Cast_12", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/transpose_93", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"128" - }, - { - "size":"1" - } - ] - } - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_17", - "op":"Const", - "attr":{ - "dtype":{ - "type":"DT_FLOAT" - }, - "value":{ - "tensor":{ - "dtype":"DT_FLOAT", - "tensorShape":{ - "dim":[ - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - }, - { - "size":"1" - } - ] - } - } - } - } - }, - { - "name":"StatefulPartitionedCall/clip_by_value_1", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_INT32", - "tensorShape":{ - "dim":[ - { - "size":"4" - } - ] - } - } - }, - "dtype":{ - "type":"DT_INT32" - } - } - }, - { - "name":"StatefulPartitionedCall/add_1", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_INT32", - "tensorShape":{ - "dim":[ - { - "size":"4" - } - ] - } - } - }, - "dtype":{ - "type":"DT_INT32" - } - } - }, - { - "name":"StatefulPartitionedCall/ones_like_1", - "op":"Const", - "attr":{ - "value":{ - "tensor":{ - "dtype":"DT_INT32", - "tensorShape":{ - "dim":[ - { - "size":"4" - } - ] - } - } - }, - "dtype":{ - "type":"DT_INT32" - } - } - }, - { - "name":"x", - "op":"Placeholder", - "attr":{ - "shape":{ - "shape":{ - "dim":[ - { - "size":"6" - }, - { - "size":"7" - } - ] - } - }, - "dtype":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Reshape_1", - "op":"Reshape", - "input":[ - "x", - "StatefulPartitionedCall/Reshape_1/shape" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - }, - "Tshape":{ - "type":"DT_INT32" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Reshape_1", - "StatefulPartitionedCall/transpose_3" - ], - "device":"/device:CPU:0", - "attr":{ - "data_format":{ - "s":"TkhXQw==" - }, - "explicit_paddings":{ - "list":{ - - } - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "padding":{ - "s":"VkFMSUQ=" - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Sub", - "op":"Add", - "input":[ - "StatefulPartitionedCall/Conv2D", - "StatefulPartitionedCall/AddV2" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_1", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub", - "StatefulPartitionedCall/Mul" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_2", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_1", - "StatefulPartitionedCall/jax2tf_arg_3/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Sub_2", - "op":"Sub", - "input":[ - "StatefulPartitionedCall/AddV2_2", - "StatefulPartitionedCall/jax2tf_arg_16/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_5", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub_2", - "StatefulPartitionedCall/Mul_4" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_7", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_5", - "StatefulPartitionedCall/jax2tf_arg_15/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Maximum_1", - "op":"Maximum", - "input":[ - "StatefulPartitionedCall/AddV2_7", - "StatefulPartitionedCall/Cast_1" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_2", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Maximum_1", - "StatefulPartitionedCall/transpose_15" - ], - "device":"/device:CPU:0", - "attr":{ - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "data_format":{ - "s":"TkhXQw==" - }, - "padding":{ - "s":"U0FNRQ==" - }, - "T":{ - "type":"DT_FLOAT" - }, - "explicit_paddings":{ - "list":{ - - } - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_9", - "op":"AddN", - "input":[ - "StatefulPartitionedCall/AddV2_2", - "StatefulPartitionedCall/Conv2D_2" - ], - "attr":{ - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - }, - "N":{ - "i":"2" - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_9", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/BroadcastTo_4", - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_9" - ], - "attr":{ - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Sub_4", - "op":"Sub", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_9", - "StatefulPartitionedCall/jax2tf_arg_32/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_9", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub_4", - "StatefulPartitionedCall/Mul_8" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_14", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_9", - "StatefulPartitionedCall/jax2tf_arg_31/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Maximum_3", - "op":"Maximum", - "input":[ - "StatefulPartitionedCall/AddV2_14", - "StatefulPartitionedCall/Cast_3" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_4", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Maximum_3", - "StatefulPartitionedCall/transpose_27" - ], - "device":"/device:CPU:0", - "attr":{ - "data_format":{ - "s":"TkhXQw==" - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "T":{ - "type":"DT_FLOAT" - }, - "explicit_paddings":{ - "list":{ - - } - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "padding":{ - "s":"U0FNRQ==" - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_16", - "op":"AddN", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_9", - "StatefulPartitionedCall/Conv2D_4" - ], - "attr":{ - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - }, - "N":{ - "i":"2" - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_16", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/BroadcastTo_6", - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_16" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - }, - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - } - } - }, - { - "name":"StatefulPartitionedCall/Sub_6", - "op":"Sub", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_16", - "StatefulPartitionedCall/jax2tf_arg_48/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_13", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub_6", - "StatefulPartitionedCall/Mul_12" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_21", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_13", - "StatefulPartitionedCall/jax2tf_arg_47/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Maximum_5", - "op":"Maximum", - "input":[ - "StatefulPartitionedCall/AddV2_21", - "StatefulPartitionedCall/Cast_5" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_6", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Maximum_5", - "StatefulPartitionedCall/transpose_39" - ], - "device":"/device:CPU:0", - "attr":{ - "padding":{ - "s":"U0FNRQ==" - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "explicit_paddings":{ - "list":{ - - } - }, - "T":{ - "type":"DT_FLOAT" - }, - "data_format":{ - "s":"TkhXQw==" - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_23", - "op":"AddN", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_16", - "StatefulPartitionedCall/Conv2D_6" - ], - "attr":{ - "N":{ - "i":"2" - }, - "T":{ - "type":"DT_FLOAT" - }, - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_23", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/BroadcastTo_8", - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_23" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - }, - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - } - } - }, - { - "name":"StatefulPartitionedCall/Sub_8", - "op":"Sub", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_23", - "StatefulPartitionedCall/jax2tf_arg_64/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_17", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub_8", - "StatefulPartitionedCall/Mul_16" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_28", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_17", - "StatefulPartitionedCall/jax2tf_arg_63/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Maximum_7", - "op":"Maximum", - "input":[ - "StatefulPartitionedCall/AddV2_28", - "StatefulPartitionedCall/Cast_7" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_8", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Maximum_7", - "StatefulPartitionedCall/transpose_51" - ], - "device":"/device:CPU:0", - "attr":{ - "use_cudnn_on_gpu":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "explicit_paddings":{ - "list":{ - - } - }, - "padding":{ - "s":"U0FNRQ==" - }, - "data_format":{ - "s":"TkhXQw==" - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_30", - "op":"AddN", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_23", - "StatefulPartitionedCall/Conv2D_8" - ], - "attr":{ - "N":{ - "i":"2" - }, - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_30", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/BroadcastTo_10", - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_30" - ], - "attr":{ - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Sub_10", - "op":"Sub", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_30", - "StatefulPartitionedCall/jax2tf_arg_80/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_21", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub_10", - "StatefulPartitionedCall/Mul_20" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_35", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_21", - "StatefulPartitionedCall/jax2tf_arg_79/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Maximum_9", - "op":"Maximum", - "input":[ - "StatefulPartitionedCall/AddV2_35", - "StatefulPartitionedCall/Cast_9" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_10", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Maximum_9", - "StatefulPartitionedCall/transpose_63" - ], - "device":"/device:CPU:0", - "attr":{ - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - }, - "data_format":{ - "s":"TkhXQw==" - }, - "explicit_paddings":{ - "list":{ - - } - }, - "padding":{ - "s":"U0FNRQ==" - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_37", - "op":"AddN", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_30", - "StatefulPartitionedCall/Conv2D_10" - ], - "attr":{ - "N":{ - "i":"2" - }, - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_37", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/BroadcastTo_12", - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_37" - ], - "attr":{ - "_grappler_ArithmeticOptimizer_AddOpsRewriteStage":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_11", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_37", - "StatefulPartitionedCall/transpose_69" - ], - "device":"/device:CPU:0", - "attr":{ - "data_format":{ - "s":"TkhXQw==" - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "explicit_paddings":{ - "list":{ - - } - }, - "T":{ - "type":"DT_FLOAT" - }, - "padding":{ - "s":"VkFMSUQ=" - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_13", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_37", - "StatefulPartitionedCall/transpose_81" - ], - "device":"/device:CPU:0", - "attr":{ - "T":{ - "type":"DT_FLOAT" - }, - "data_format":{ - "s":"TkhXQw==" - }, - "padding":{ - "s":"VkFMSUQ=" - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "explicit_paddings":{ - "list":{ - - } - } - } - }, - { - "name":"StatefulPartitionedCall/Sub_11", - "op":"Add", - "input":[ - "StatefulPartitionedCall/Conv2D_11", - "StatefulPartitionedCall/AddV2_38" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Sub_12", - "op":"Add", - "input":[ - "StatefulPartitionedCall/Conv2D_13", - "StatefulPartitionedCall/AddV2_42" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_23", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub_11", - "StatefulPartitionedCall/Mul_22" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_25", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub_12", - "StatefulPartitionedCall/Mul_24" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_40", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_23", - "StatefulPartitionedCall/jax2tf_arg_91/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_44", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_25", - "StatefulPartitionedCall/jax2tf_arg_101/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Maximum_10", - "op":"Maximum", - "input":[ - "StatefulPartitionedCall/AddV2_40", - "StatefulPartitionedCall/Cast_10" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Maximum_11", - "op":"Maximum", - "input":[ - "StatefulPartitionedCall/AddV2_44", - "StatefulPartitionedCall/Cast_11" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_12", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Maximum_10", - "StatefulPartitionedCall/transpose_75" - ], - "device":"/device:CPU:0", - "attr":{ - "padding":{ - "s":"VkFMSUQ=" - }, - "data_format":{ - "s":"TkhXQw==" - }, - "explicit_paddings":{ - "list":{ - - } - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_14", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Maximum_11", - "StatefulPartitionedCall/transpose_87" - ], - "device":"/device:CPU:0", - "attr":{ - "use_cudnn_on_gpu":{ - "b":true - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "padding":{ - "s":"VkFMSUQ=" - }, - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "explicit_paddings":{ - "list":{ - - } - }, - "data_format":{ - "s":"TkhXQw==" - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_41", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Conv2D_12", - "StatefulPartitionedCall/BroadcastTo_14" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Sub_13", - "op":"Add", - "input":[ - "StatefulPartitionedCall/Conv2D_14", - "StatefulPartitionedCall/AddV2_45" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/StridedSlice", - "op":"StridedSlice", - "input":[ - "StatefulPartitionedCall/AddV2_41", - "StatefulPartitionedCall/clip_by_value", - "StatefulPartitionedCall/add", - "StatefulPartitionedCall/ones_like" - ], - "attr":{ - "new_axis_mask":{ - "i":"0" - }, - "shrink_axis_mask":{ - "i":"7" - }, - "begin_mask":{ - "i":"0" - }, - "end_mask":{ - "i":"0" - }, - "T":{ - "type":"DT_FLOAT" - }, - "Index":{ - "type":"DT_INT32" - }, - "ellipsis_mask":{ - "i":"0" - } - } - }, - { - "name":"StatefulPartitionedCall/Mul_27", - "op":"Mul", - "input":[ - "StatefulPartitionedCall/Sub_13", - "StatefulPartitionedCall/Mul_26" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"Identity", - "op":"Identity", - "input":[ - "StatefulPartitionedCall/StridedSlice" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_47", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Mul_27", - "StatefulPartitionedCall/jax2tf_arg_109/ReadVariableOp" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Maximum_12", - "op":"Maximum", - "input":[ - "StatefulPartitionedCall/AddV2_47", - "StatefulPartitionedCall/Cast_12" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Conv2D_15", - "op":"Conv2D", - "input":[ - "StatefulPartitionedCall/Maximum_12", - "StatefulPartitionedCall/transpose_93" - ], - "device":"/device:CPU:0", - "attr":{ - "strides":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "data_format":{ - "s":"TkhXQw==" - }, - "use_cudnn_on_gpu":{ - "b":true - }, - "explicit_paddings":{ - "list":{ - - } - }, - "padding":{ - "s":"VkFMSUQ=" - }, - "dilations":{ - "list":{ - "i":[ - "1", - "1", - "1", - "1" - ] - } - }, - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/AddV2_48", - "op":"AddV2", - "input":[ - "StatefulPartitionedCall/Conv2D_15", - "StatefulPartitionedCall/BroadcastTo_17" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/Tanh", - "op":"Tanh", - "input":[ - "StatefulPartitionedCall/AddV2_48" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - }, - { - "name":"StatefulPartitionedCall/StridedSlice_1", - "op":"StridedSlice", - "input":[ - "StatefulPartitionedCall/Tanh", - "StatefulPartitionedCall/clip_by_value_1", - "StatefulPartitionedCall/add_1", - "StatefulPartitionedCall/ones_like_1" - ], - "attr":{ - "ellipsis_mask":{ - "i":"0" - }, - "new_axis_mask":{ - "i":"0" - }, - "end_mask":{ - "i":"0" - }, - "shrink_axis_mask":{ - "i":"15" - }, - "T":{ - "type":"DT_FLOAT" - }, - "begin_mask":{ - "i":"0" - }, - "Index":{ - "type":"DT_INT32" - } - } - }, - { - "name":"Identity_1", - "op":"Identity", - "input":[ - "StatefulPartitionedCall/StridedSlice_1" - ], - "attr":{ - "T":{ - "type":"DT_FLOAT" - } - } - } - ], - "library":{ - - }, - "versions":{ - "producer":1149 - } - }, - "weightsManifest":[ - { - "paths":[ - "group1-shard1of2.bin", - "group1-shard2of2.bin" - ], - "weights":[ - { - "name":"StatefulPartitionedCall/transpose_69", - "shape":[ - 1, - 1, - 128, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/AddV2_38", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul_22", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_91/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Cast_10", - "shape":[ - - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_75", - "shape":[ - 6, - 7, - 128, - 7 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_14", - "shape":[ - 1, - 1, - 1, - 7 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/clip_by_value", - "shape":[ - 4 - ], - "dtype":"int32" - }, - { - "name":"StatefulPartitionedCall/add", - "shape":[ - 4 - ], - "dtype":"int32" - }, - { - "name":"StatefulPartitionedCall/ones_like", - "shape":[ - 4 - ], - "dtype":"int32" - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_12", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_10", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_8", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_6", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_4", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Reshape_1/shape", - "shape":[ - 4 - ], - "dtype":"int32" - }, - { - "name":"StatefulPartitionedCall/transpose_3", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/AddV2", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_3/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_16/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul_4", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_15/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Cast_1", - "shape":[ - - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_15", - "shape":[ - 3, - 3, - 128, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_32/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul_8", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_31/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Cast_3", - "shape":[ - - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_27", - "shape":[ - 3, - 3, - 128, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_48/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul_12", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_47/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Cast_5", - "shape":[ - - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_39", - "shape":[ - 3, - 3, - 128, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_64/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul_16", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_63/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Cast_7", - "shape":[ - - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_51", - "shape":[ - 3, - 3, - 128, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_80/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul_20", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_79/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Cast_9", - "shape":[ - - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_63", - "shape":[ - 3, - 3, - 128, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_81", - "shape":[ - 1, - 1, - 128, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/AddV2_42", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul_24", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_101/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Cast_11", - "shape":[ - - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_87", - "shape":[ - 6, - 7, - 128, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/AddV2_45", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Mul_26", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/jax2tf_arg_109/ReadVariableOp", - "shape":[ - 1, - 1, - 1, - 128 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/Cast_12", - "shape":[ - - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/transpose_93", - "shape":[ - 1, - 1, - 128, - 1 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/BroadcastTo_17", - "shape":[ - 1, - 1, - 1, - 1 - ], - "dtype":"float32" - }, - { - "name":"StatefulPartitionedCall/clip_by_value_1", - "shape":[ - 4 - ], - "dtype":"int32" - }, - { - "name":"StatefulPartitionedCall/add_1", - "shape":[ - 4 - ], - "dtype":"int32" - }, - { - "name":"StatefulPartitionedCall/ones_like_1", - "shape":[ - 4 - ], - "dtype":"int32" - } - ] - } - ] -} \ No newline at end of file +{"format": "graph-model", "generatedBy": "2.10.0-dev20220602", "convertedBy": "TensorFlow.js Converter v3.18.0", "signature": {"inputs": {"x": {"name": "x:0", "dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "6"}, {"size": "7"}]}}}, "outputs": {"output_1": {"name": "Identity_1:0", "dtype": "DT_FLOAT", "tensorShape": {}}, "output_0": {"name": "Identity:0", "dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "7"}]}}}}, "modelTopology": {"node": [{"name": "StatefulPartitionedCall/transpose_69", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "128"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/AddV2_38", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_22", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_91/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Cast_10", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {}}}}}, {"name": "StatefulPartitionedCall/transpose_75", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "6"}, {"size": "7"}, {"size": "128"}, {"size": "7"}]}}}}}, {"name": "StatefulPartitionedCall/BroadcastTo_14", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "7"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/clip_by_value", "op": "Const", "attr": {"dtype": {"type": "DT_INT32"}, "value": {"tensor": {"dtype": "DT_INT32", "tensorShape": {"dim": [{"size": "4"}]}}}}}, {"name": "StatefulPartitionedCall/add", "op": "Const", "attr": {"dtype": {"type": "DT_INT32"}, "value": {"tensor": {"dtype": "DT_INT32", "tensorShape": {"dim": [{"size": "4"}]}}}}}, {"name": "StatefulPartitionedCall/ones_like", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_INT32", "tensorShape": {"dim": [{"size": "4"}]}}}, "dtype": {"type": "DT_INT32"}}}, {"name": "StatefulPartitionedCall/BroadcastTo_12", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/BroadcastTo_10", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/BroadcastTo_8", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/BroadcastTo_6", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/BroadcastTo_4", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Reshape_1/shape", "op": "Const", "attr": {"dtype": {"type": "DT_INT32"}, "value": {"tensor": {"dtype": "DT_INT32", "tensorShape": {"dim": [{"size": "4"}]}}}}}, {"name": "StatefulPartitionedCall/transpose_3", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Mul", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_3/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_16/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Mul_4", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_15/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Cast_1", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {}}}}}, {"name": "StatefulPartitionedCall/transpose_15", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "3"}, {"size": "3"}, {"size": "128"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_32/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Mul_8", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_31/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Cast_3", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {}}}}}, {"name": "StatefulPartitionedCall/transpose_27", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "3"}, {"size": "3"}, {"size": "128"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_48/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Mul_12", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_47/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Cast_5", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {}}}}}, {"name": "StatefulPartitionedCall/transpose_39", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "3"}, {"size": "3"}, {"size": "128"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_64/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Mul_16", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_63/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Cast_7", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {}}}}}, {"name": "StatefulPartitionedCall/transpose_51", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "3"}, {"size": "3"}, {"size": "128"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_80/ReadVariableOp", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_20", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_79/ReadVariableOp", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Cast_9", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {}}}}}, {"name": "StatefulPartitionedCall/transpose_63", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "3"}, {"size": "3"}, {"size": "128"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/transpose_81", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "128"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_42", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Mul_24", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_101/ReadVariableOp", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Cast_11", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/transpose_87", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "6"}, {"size": "7"}, {"size": "128"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_45", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/Mul_26", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}}}, {"name": "StatefulPartitionedCall/jax2tf_arg_109/ReadVariableOp", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "128"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Cast_12", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/transpose_93", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "128"}, {"size": "1"}]}}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/BroadcastTo_17", "op": "Const", "attr": {"dtype": {"type": "DT_FLOAT"}, "value": {"tensor": {"dtype": "DT_FLOAT", "tensorShape": {"dim": [{"size": "1"}, {"size": "1"}, {"size": "1"}, {"size": "1"}]}}}}}, {"name": "StatefulPartitionedCall/clip_by_value_1", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_INT32", "tensorShape": {"dim": [{"size": "4"}]}}}, "dtype": {"type": "DT_INT32"}}}, {"name": "StatefulPartitionedCall/add_1", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_INT32", "tensorShape": {"dim": [{"size": "4"}]}}}, "dtype": {"type": "DT_INT32"}}}, {"name": "StatefulPartitionedCall/ones_like_1", "op": "Const", "attr": {"value": {"tensor": {"dtype": "DT_INT32", "tensorShape": {"dim": [{"size": "4"}]}}}, "dtype": {"type": "DT_INT32"}}}, {"name": "x", "op": "Placeholder", "attr": {"shape": {"shape": {"dim": [{"size": "6"}, {"size": "7"}]}}, "dtype": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Reshape_1", "op": "Reshape", "input": ["x", "StatefulPartitionedCall/Reshape_1/shape"], "attr": {"T": {"type": "DT_FLOAT"}, "Tshape": {"type": "DT_INT32"}}}, {"name": "StatefulPartitionedCall/Conv2D", "op": "Conv2D", "input": ["StatefulPartitionedCall/Reshape_1", "StatefulPartitionedCall/transpose_3"], "device": "/device:CPU:0", "attr": {"data_format": {"s": "TkhXQw=="}, "explicit_paddings": {"list": {}}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}, "use_cudnn_on_gpu": {"b": true}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "padding": {"s": "VkFMSUQ="}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Sub", "op": "Add", "input": ["StatefulPartitionedCall/Conv2D", "StatefulPartitionedCall/AddV2"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_1", "op": "Mul", "input": ["StatefulPartitionedCall/Sub", "StatefulPartitionedCall/Mul"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_2", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_1", "StatefulPartitionedCall/jax2tf_arg_3/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Sub_2", "op": "Sub", "input": ["StatefulPartitionedCall/AddV2_2", "StatefulPartitionedCall/jax2tf_arg_16/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_5", "op": "Mul", "input": ["StatefulPartitionedCall/Sub_2", "StatefulPartitionedCall/Mul_4"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_7", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_5", "StatefulPartitionedCall/jax2tf_arg_15/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Maximum_1", "op": "Maximum", "input": ["StatefulPartitionedCall/AddV2_7", "StatefulPartitionedCall/Cast_1"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_2", "op": "Conv2D", "input": ["StatefulPartitionedCall/Maximum_1", "StatefulPartitionedCall/transpose_15"], "device": "/device:CPU:0", "attr": {"dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "data_format": {"s": "TkhXQw=="}, "padding": {"s": "U0FNRQ=="}, "T": {"type": "DT_FLOAT"}, "explicit_paddings": {"list": {}}, "use_cudnn_on_gpu": {"b": true}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_9", "op": "AddN", "input": ["StatefulPartitionedCall/AddV2_2", "StatefulPartitionedCall/Conv2D_2"], "attr": {"_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}, "N": {"i": "2"}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_9", "op": "AddV2", "input": ["StatefulPartitionedCall/BroadcastTo_4", "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_9"], "attr": {"_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Sub_4", "op": "Sub", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_9", "StatefulPartitionedCall/jax2tf_arg_32/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_9", "op": "Mul", "input": ["StatefulPartitionedCall/Sub_4", "StatefulPartitionedCall/Mul_8"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_14", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_9", "StatefulPartitionedCall/jax2tf_arg_31/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Maximum_3", "op": "Maximum", "input": ["StatefulPartitionedCall/AddV2_14", "StatefulPartitionedCall/Cast_3"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_4", "op": "Conv2D", "input": ["StatefulPartitionedCall/Maximum_3", "StatefulPartitionedCall/transpose_27"], "device": "/device:CPU:0", "attr": {"data_format": {"s": "TkhXQw=="}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "T": {"type": "DT_FLOAT"}, "explicit_paddings": {"list": {}}, "use_cudnn_on_gpu": {"b": true}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}, "padding": {"s": "U0FNRQ=="}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_16", "op": "AddN", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_9", "StatefulPartitionedCall/Conv2D_4"], "attr": {"_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}, "T": {"type": "DT_FLOAT"}, "N": {"i": "2"}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_16", "op": "AddV2", "input": ["StatefulPartitionedCall/BroadcastTo_6", "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_16"], "attr": {"T": {"type": "DT_FLOAT"}, "_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}}}, {"name": "StatefulPartitionedCall/Sub_6", "op": "Sub", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_16", "StatefulPartitionedCall/jax2tf_arg_48/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_13", "op": "Mul", "input": ["StatefulPartitionedCall/Sub_6", "StatefulPartitionedCall/Mul_12"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_21", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_13", "StatefulPartitionedCall/jax2tf_arg_47/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Maximum_5", "op": "Maximum", "input": ["StatefulPartitionedCall/AddV2_21", "StatefulPartitionedCall/Cast_5"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_6", "op": "Conv2D", "input": ["StatefulPartitionedCall/Maximum_5", "StatefulPartitionedCall/transpose_39"], "device": "/device:CPU:0", "attr": {"padding": {"s": "U0FNRQ=="}, "use_cudnn_on_gpu": {"b": true}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "explicit_paddings": {"list": {}}, "T": {"type": "DT_FLOAT"}, "data_format": {"s": "TkhXQw=="}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_23", "op": "AddN", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_16", "StatefulPartitionedCall/Conv2D_6"], "attr": {"N": {"i": "2"}, "T": {"type": "DT_FLOAT"}, "_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_23", "op": "AddV2", "input": ["StatefulPartitionedCall/BroadcastTo_8", "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_23"], "attr": {"T": {"type": "DT_FLOAT"}, "_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}}}, {"name": "StatefulPartitionedCall/Sub_8", "op": "Sub", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_23", "StatefulPartitionedCall/jax2tf_arg_64/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_17", "op": "Mul", "input": ["StatefulPartitionedCall/Sub_8", "StatefulPartitionedCall/Mul_16"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_28", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_17", "StatefulPartitionedCall/jax2tf_arg_63/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Maximum_7", "op": "Maximum", "input": ["StatefulPartitionedCall/AddV2_28", "StatefulPartitionedCall/Cast_7"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_8", "op": "Conv2D", "input": ["StatefulPartitionedCall/Maximum_7", "StatefulPartitionedCall/transpose_51"], "device": "/device:CPU:0", "attr": {"use_cudnn_on_gpu": {"b": true}, "T": {"type": "DT_FLOAT"}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "explicit_paddings": {"list": {}}, "padding": {"s": "U0FNRQ=="}, "data_format": {"s": "TkhXQw=="}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_30", "op": "AddN", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_23", "StatefulPartitionedCall/Conv2D_8"], "attr": {"N": {"i": "2"}, "_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_30", "op": "AddV2", "input": ["StatefulPartitionedCall/BroadcastTo_10", "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_30"], "attr": {"_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Sub_10", "op": "Sub", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_30", "StatefulPartitionedCall/jax2tf_arg_80/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_21", "op": "Mul", "input": ["StatefulPartitionedCall/Sub_10", "StatefulPartitionedCall/Mul_20"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_35", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_21", "StatefulPartitionedCall/jax2tf_arg_79/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Maximum_9", "op": "Maximum", "input": ["StatefulPartitionedCall/AddV2_35", "StatefulPartitionedCall/Cast_9"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_10", "op": "Conv2D", "input": ["StatefulPartitionedCall/Maximum_9", "StatefulPartitionedCall/transpose_63"], "device": "/device:CPU:0", "attr": {"strides": {"list": {"i": ["1", "1", "1", "1"]}}, "use_cudnn_on_gpu": {"b": true}, "T": {"type": "DT_FLOAT"}, "data_format": {"s": "TkhXQw=="}, "explicit_paddings": {"list": {}}, "padding": {"s": "U0FNRQ=="}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_37", "op": "AddN", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_30", "StatefulPartitionedCall/Conv2D_10"], "attr": {"N": {"i": "2"}, "_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_37", "op": "AddV2", "input": ["StatefulPartitionedCall/BroadcastTo_12", "StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_Leaf_1_AddV2_37"], "attr": {"_grappler_ArithmeticOptimizer_AddOpsRewriteStage": {"b": true}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_11", "op": "Conv2D", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_37", "StatefulPartitionedCall/transpose_69"], "device": "/device:CPU:0", "attr": {"data_format": {"s": "TkhXQw=="}, "use_cudnn_on_gpu": {"b": true}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "explicit_paddings": {"list": {}}, "T": {"type": "DT_FLOAT"}, "padding": {"s": "VkFMSUQ="}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}}}, {"name": "StatefulPartitionedCall/Conv2D_13", "op": "Conv2D", "input": ["StatefulPartitionedCall/ArithmeticOptimizer/AddOpsRewrite_AddV2_37", "StatefulPartitionedCall/transpose_81"], "device": "/device:CPU:0", "attr": {"T": {"type": "DT_FLOAT"}, "data_format": {"s": "TkhXQw=="}, "padding": {"s": "VkFMSUQ="}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "use_cudnn_on_gpu": {"b": true}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}, "explicit_paddings": {"list": {}}}}, {"name": "StatefulPartitionedCall/Sub_11", "op": "Add", "input": ["StatefulPartitionedCall/Conv2D_11", "StatefulPartitionedCall/AddV2_38"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Sub_12", "op": "Add", "input": ["StatefulPartitionedCall/Conv2D_13", "StatefulPartitionedCall/AddV2_42"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_23", "op": "Mul", "input": ["StatefulPartitionedCall/Sub_11", "StatefulPartitionedCall/Mul_22"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Mul_25", "op": "Mul", "input": ["StatefulPartitionedCall/Sub_12", "StatefulPartitionedCall/Mul_24"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_40", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_23", "StatefulPartitionedCall/jax2tf_arg_91/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_44", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_25", "StatefulPartitionedCall/jax2tf_arg_101/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Maximum_10", "op": "Maximum", "input": ["StatefulPartitionedCall/AddV2_40", "StatefulPartitionedCall/Cast_10"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Maximum_11", "op": "Maximum", "input": ["StatefulPartitionedCall/AddV2_44", "StatefulPartitionedCall/Cast_11"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_12", "op": "Conv2D", "input": ["StatefulPartitionedCall/Maximum_10", "StatefulPartitionedCall/transpose_75"], "device": "/device:CPU:0", "attr": {"padding": {"s": "VkFMSUQ="}, "data_format": {"s": "TkhXQw=="}, "explicit_paddings": {"list": {}}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}, "use_cudnn_on_gpu": {"b": true}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_14", "op": "Conv2D", "input": ["StatefulPartitionedCall/Maximum_11", "StatefulPartitionedCall/transpose_87"], "device": "/device:CPU:0", "attr": {"use_cudnn_on_gpu": {"b": true}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "padding": {"s": "VkFMSUQ="}, "strides": {"list": {"i": ["1", "1", "1", "1"]}}, "explicit_paddings": {"list": {}}, "data_format": {"s": "TkhXQw=="}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_41", "op": "AddV2", "input": ["StatefulPartitionedCall/Conv2D_12", "StatefulPartitionedCall/BroadcastTo_14"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Sub_13", "op": "Add", "input": ["StatefulPartitionedCall/Conv2D_14", "StatefulPartitionedCall/AddV2_45"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/StridedSlice", "op": "StridedSlice", "input": ["StatefulPartitionedCall/AddV2_41", "StatefulPartitionedCall/clip_by_value", "StatefulPartitionedCall/add", "StatefulPartitionedCall/ones_like"], "attr": {"new_axis_mask": {"i": "0"}, "shrink_axis_mask": {"i": "7"}, "begin_mask": {"i": "0"}, "end_mask": {"i": "0"}, "T": {"type": "DT_FLOAT"}, "Index": {"type": "DT_INT32"}, "ellipsis_mask": {"i": "0"}}}, {"name": "StatefulPartitionedCall/Mul_27", "op": "Mul", "input": ["StatefulPartitionedCall/Sub_13", "StatefulPartitionedCall/Mul_26"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "Identity", "op": "Identity", "input": ["StatefulPartitionedCall/StridedSlice"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_47", "op": "AddV2", "input": ["StatefulPartitionedCall/Mul_27", "StatefulPartitionedCall/jax2tf_arg_109/ReadVariableOp"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Maximum_12", "op": "Maximum", "input": ["StatefulPartitionedCall/AddV2_47", "StatefulPartitionedCall/Cast_12"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Conv2D_15", "op": "Conv2D", "input": ["StatefulPartitionedCall/Maximum_12", "StatefulPartitionedCall/transpose_93"], "device": "/device:CPU:0", "attr": {"strides": {"list": {"i": ["1", "1", "1", "1"]}}, "data_format": {"s": "TkhXQw=="}, "use_cudnn_on_gpu": {"b": true}, "explicit_paddings": {"list": {}}, "padding": {"s": "VkFMSUQ="}, "dilations": {"list": {"i": ["1", "1", "1", "1"]}}, "T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/AddV2_48", "op": "AddV2", "input": ["StatefulPartitionedCall/Conv2D_15", "StatefulPartitionedCall/BroadcastTo_17"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/Tanh", "op": "Tanh", "input": ["StatefulPartitionedCall/AddV2_48"], "attr": {"T": {"type": "DT_FLOAT"}}}, {"name": "StatefulPartitionedCall/StridedSlice_1", "op": "StridedSlice", "input": ["StatefulPartitionedCall/Tanh", "StatefulPartitionedCall/clip_by_value_1", "StatefulPartitionedCall/add_1", "StatefulPartitionedCall/ones_like_1"], "attr": {"ellipsis_mask": {"i": "0"}, "new_axis_mask": {"i": "0"}, "end_mask": {"i": "0"}, "shrink_axis_mask": {"i": "15"}, "T": {"type": "DT_FLOAT"}, "begin_mask": {"i": "0"}, "Index": {"type": "DT_INT32"}}}, {"name": "Identity_1", "op": "Identity", "input": ["StatefulPartitionedCall/StridedSlice_1"], "attr": {"T": {"type": "DT_FLOAT"}}}], "library": {}, "versions": {"producer": 1149}}, "weightsManifest": [{"paths": ["group1-shard1of2.bin", "group1-shard2of2.bin"], "weights": [{"name": "StatefulPartitionedCall/transpose_69", "shape": [1, 1, 128, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/AddV2_38", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul_22", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_91/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Cast_10", "shape": [], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_75", "shape": [6, 7, 128, 7], "dtype": "float32"}, {"name": "StatefulPartitionedCall/BroadcastTo_14", "shape": [1, 1, 1, 7], "dtype": "float32"}, {"name": "StatefulPartitionedCall/clip_by_value", "shape": [4], "dtype": "int32"}, {"name": "StatefulPartitionedCall/add", "shape": [4], "dtype": "int32"}, {"name": "StatefulPartitionedCall/ones_like", "shape": [4], "dtype": "int32"}, {"name": "StatefulPartitionedCall/BroadcastTo_12", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/BroadcastTo_10", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/BroadcastTo_8", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/BroadcastTo_6", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/BroadcastTo_4", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Reshape_1/shape", "shape": [4], "dtype": "int32"}, {"name": "StatefulPartitionedCall/transpose_3", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/AddV2", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_3/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_16/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul_4", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_15/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Cast_1", "shape": [], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_15", "shape": [3, 3, 128, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_32/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul_8", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_31/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Cast_3", "shape": [], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_27", "shape": [3, 3, 128, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_48/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul_12", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_47/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Cast_5", "shape": [], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_39", "shape": [3, 3, 128, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_64/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul_16", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_63/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Cast_7", "shape": [], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_51", "shape": [3, 3, 128, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_80/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul_20", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_79/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Cast_9", "shape": [], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_63", "shape": [3, 3, 128, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_81", "shape": [1, 1, 128, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/AddV2_42", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul_24", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_101/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Cast_11", "shape": [], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_87", "shape": [6, 7, 128, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/AddV2_45", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Mul_26", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/jax2tf_arg_109/ReadVariableOp", "shape": [1, 1, 1, 128], "dtype": "float32"}, {"name": "StatefulPartitionedCall/Cast_12", "shape": [], "dtype": "float32"}, {"name": "StatefulPartitionedCall/transpose_93", "shape": [1, 1, 128, 1], "dtype": "float32"}, {"name": "StatefulPartitionedCall/BroadcastTo_17", "shape": [1, 1, 1, 1], "dtype": "float32"}, {"name": "StatefulPartitionedCall/clip_by_value_1", "shape": [4], "dtype": "int32"}, {"name": "StatefulPartitionedCall/add_1", "shape": [4], "dtype": "int32"}, {"name": "StatefulPartitionedCall/ones_like_1", "shape": [4], "dtype": "int32"}]}]} \ No newline at end of file