Skip to content

Instantly share code, notes, and snippets.

@letalvoj

letalvoj/TF-nGraph-PlaidML.ipynb

Created November 24, 2019 13:57
Show Gist options
  • Save letalvoj/d602e22e9ae04c31afc89d58f4977075 to your computer and use it in GitHub Desktop.
Save letalvoj/d602e22e9ae04c31afc89d58f4977075 to your computer and use it in GitHub Desktop.
Download ZIP
Error while trying to set up tensorflow with nGraph backend with PlaidML backend.
Display the source blob
Display the rendered blob
Raw
TF-nGraph-PlaidML.ipynb
{
"cells": [
{
"cell_type": "code",
"execution_count": 2,
"metadata": {
"ExecuteTime": {
"end_time": "2019-11-24T13:51:48.583699Z",
"start_time": "2019-11-24T13:51:47.017524Z"
},
"scrolled": true
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"['CPU', 'INTERPRETER', 'NOP', 'PLAIDML']\n"
]
}
],
"source": [
"import ngraph_bridge\n",
"print(ngraph_bridge.list_backends())"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {
"ExecuteTime": {
"end_time": "2019-11-24T13:51:48.661715Z",
"start_time": "2019-11-24T13:51:48.585719Z"
}
},
"outputs": [],
"source": [
"ngraph_bridge.set_backend('PLAIDML')"
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {
"ExecuteTime": {
"end_time": "2019-11-24T13:51:48.666569Z",
"start_time": "2019-11-24T13:51:48.663965Z"
}
},
"outputs": [],
"source": [
"import tensorflow as tf"
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {
"ExecuteTime": {
"end_time": "2019-11-24T13:51:48.672412Z",
"start_time": "2019-11-24T13:51:48.669082Z"
}
},
"outputs": [],
"source": [
"__config = tf.ConfigProto(allow_soft_placement=True,\n",
" log_device_placement=True,\n",
" inter_op_parallelism_threads=4)\n",
"\n",
"config_ngraph_enabled = ngraph_bridge.update_config(__config)"
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {
"ExecuteTime": {
"end_time": "2019-11-24T13:51:48.917820Z",
"start_time": "2019-11-24T13:51:48.675066Z"
}
},
"outputs": [],
"source": [
"(x_train, y_train), (x_test, y_test) = tf.keras.datasets.mnist.load_data()"
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {
"ExecuteTime": {
"end_time": "2019-11-24T13:51:48.926653Z",
"start_time": "2019-11-24T13:51:48.920258Z"
}
},
"outputs": [],
"source": [
"session = tf.Session(config=config_ngraph_enabled)\n",
"tf.keras.backend.set_session(session)"
]
},
{
"cell_type": "code",
"execution_count": 8,
"metadata": {
"ExecuteTime": {
"end_time": "2019-11-24T13:51:49.120846Z",
"start_time": "2019-11-24T13:51:48.928569Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"WARNING:tensorflow:From ./venv/lib/python3.7/site-packages/tensorflow/python/ops/init_ops.py:1251: calling VarianceScaling.__init__ (from tensorflow.python.ops.init_ops) with dtype is deprecated and will be removed in a future version.\n",
"Instructions for updating:\n",
"Call initializer instance with the dtype argument instead of passing it to the constructor\n"
]
}
],
"source": [
"model = tf.keras.models.Sequential([\n",
" tf.keras.layers.Flatten(input_shape=(28, 28)),\n",
" tf.keras.layers.Dense(128, activation='relu'),\n",
" tf.keras.layers.Dropout(0.2),\n",
" tf.keras.layers.Dense(10, activation='softmax')\n",
"])\n",
"\n",
"model.compile(optimizer='adam',\n",
" loss='sparse_categorical_crossentropy',\n",
" metrics=['accuracy'])"
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {
"ExecuteTime": {
"end_time": "2019-11-24T13:51:49.871254Z",
"start_time": "2019-11-24T13:51:49.123259Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch 1/20\n"
]
},
{
"ename": "InternalError",
"evalue": "Caught exception while executing nGraph computation: Second argument of index must be an integer\n\n\t [[{{node ngraph_cluster_17}}]]",
"output_type": "error",
"traceback": [
"\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
"\u001b[0;31mInternalError\u001b[0m Traceback (most recent call last)",
"\u001b[0;32m<ipython-input-9-da57676bce5f>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m()\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[0mmodel\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mfit\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mx_train\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0my_train\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mepochs\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m20\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m 2\u001b[0m \u001b[0mmodel\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mevaluate\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mx_test\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0my_test\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mverbose\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m2\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
"\u001b[0;32m~/Workspace/Personal/ngraphtf/venv/lib/python3.7/site-packages/tensorflow/python/keras/engine/training.py\u001b[0m in \u001b[0;36mfit\u001b[0;34m(self, x, y, batch_size, epochs, verbose, callbacks, validation_split, validation_data, shuffle, class_weight, sample_weight, initial_epoch, steps_per_epoch, validation_steps, validation_freq, max_queue_size, workers, use_multiprocessing, **kwargs)\u001b[0m\n\u001b[1;32m 778\u001b[0m \u001b[0mvalidation_steps\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mvalidation_steps\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 779\u001b[0m \u001b[0mvalidation_freq\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mvalidation_freq\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 780\u001b[0;31m steps_name='steps_per_epoch')\n\u001b[0m\u001b[1;32m 781\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 782\u001b[0m def evaluate(self,\n",
"\u001b[0;32m~/Workspace/Personal/ngraphtf/venv/lib/python3.7/site-packages/tensorflow/python/keras/engine/training_arrays.py\u001b[0m in \u001b[0;36mmodel_iteration\u001b[0;34m(model, inputs, targets, sample_weights, batch_size, epochs, verbose, callbacks, val_inputs, val_targets, val_sample_weights, shuffle, initial_epoch, steps_per_epoch, validation_steps, validation_freq, mode, validation_in_fit, prepared_feed_values_from_dataset, steps_name, **kwargs)\u001b[0m\n\u001b[1;32m 361\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 362\u001b[0m \u001b[0;31m# Get outputs.\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 363\u001b[0;31m \u001b[0mbatch_outs\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mf\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mins_batch\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m 364\u001b[0m \u001b[0;32mif\u001b[0m \u001b[0;32mnot\u001b[0m \u001b[0misinstance\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mbatch_outs\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mlist\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 365\u001b[0m \u001b[0mbatch_outs\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;34m[\u001b[0m\u001b[0mbatch_outs\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
"\u001b[0;32m~/Workspace/Personal/ngraphtf/venv/lib/python3.7/site-packages/tensorflow/python/keras/backend.py\u001b[0m in \u001b[0;36m__call__\u001b[0;34m(self, inputs)\u001b[0m\n\u001b[1;32m 3290\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 3291\u001b[0m fetched = self._callable_fn(*array_vals,\n\u001b[0;32m-> 3292\u001b[0;31m run_metadata=self.run_metadata)\n\u001b[0m\u001b[1;32m 3293\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_call_fetch_callbacks\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mfetched\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m-\u001b[0m\u001b[0mlen\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_fetches\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 3294\u001b[0m output_structure = nest.pack_sequence_as(\n",
"\u001b[0;32m~/Workspace/Personal/ngraphtf/venv/lib/python3.7/site-packages/tensorflow/python/client/session.py\u001b[0m in \u001b[0;36m__call__\u001b[0;34m(self, *args, **kwargs)\u001b[0m\n\u001b[1;32m 1456\u001b[0m ret = tf_session.TF_SessionRunCallable(self._session._session,\n\u001b[1;32m 1457\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_handle\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0margs\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m-> 1458\u001b[0;31m run_metadata_ptr)\n\u001b[0m\u001b[1;32m 1459\u001b[0m \u001b[0;32mif\u001b[0m \u001b[0mrun_metadata\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 1460\u001b[0m \u001b[0mproto_data\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mtf_session\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mTF_GetBuffer\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mrun_metadata_ptr\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
"\u001b[0;31mInternalError\u001b[0m: Caught exception while executing nGraph computation: Second argument of index must be an integer\n\n\t [[{{node ngraph_cluster_17}}]]"
]
}
],
"source": [
"model.fit(x_train, y_train, epochs=20)\n",
"model.evaluate(x_test, y_test, verbose=2)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "ngraphtf-venv",
"language": "python",
"name": "ngraphtf-venv"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.7.5"
},
"latex_envs": {
"bibliofile": "biblio.bib",
"cite_by": "apalike",
"current_citInitial": 1,
"eqLabelWithNumbers": true,
"eqNumInitial": 0
}
},
"nbformat": 4,
"nbformat_minor": 2
}
Raw
tf_function_error_ngraph_cluster_17.json
[
{
"name": "Function_12",
"ops": [
{
"cacheable": false,
"element_type": "float",
"name": "Parameter_942",
"op": "Parameter",
"outputs": [
"Parameter_942_0"
],
"shape": [
32,
128
]
},
{
"cacheable": false,
"element_type": "float",
"name": "Parameter_943",
"op": "Parameter",
"outputs": [
"Parameter_943_0"
],
"shape": [
128,
10
]
},
{
"cacheable": false,
"element_type": "float",
"name": "Parameter_944",
"op": "Parameter",
"outputs": [
"Parameter_944_0"
],
"shape": [
10
]
},
{
"cacheable": false,
"element_type": "int32_t",
"name": "Parameter_945",
"op": "Parameter",
"outputs": [
"Parameter_945_0"
],
"shape": [
1
]
},
{
"cacheable": false,
"element_type": "int32_t",
"name": "Parameter_946",
"op": "Parameter",
"outputs": [
"Parameter_946_0"
],
"shape": []
},
{
"cacheable": false,
"element_type": "float",
"name": "Parameter_947",
"op": "Parameter",
"outputs": [
"Parameter_947_0"
],
"shape": [
32,
1
]
},
{
"friendly_name": "dense_1/BiasAdd",
"inputs": [
"Parameter_942",
"Parameter_943"
],
"name": "Dot_948",
"op": "Dot",
"outputs": [
"Dot_948_0"
],
"reduction_axes_count": 1
},
{
"axes": [
0
],
"friendly_name": "dense_1/BiasAdd",
"inputs": [
"Parameter_944"
],
"name": "Broadcast_949",
"op": "Broadcast",
"outputs": [
"Broadcast_949_0"
],
"shape": [
32,
10
]
},
{
"friendly_name": "dense_1/BiasAdd",
"inputs": [
"Dot_948",
"Broadcast_949"
],
"name": "Add_950",
"op": "Add",
"outputs": [
"Add_950_0"
]
},
{
"element_type": "int64_t",
"name": "Constant_964",
"op": "Constant",
"outputs": [
"Constant_964_0"
],
"shape": [
1
],
"value": [
"1"
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Add_950",
"Constant_964"
],
"name": "Max_965",
"op": "Max",
"outputs": [
"Max_965_0"
],
"reduction_axes": [
1
]
},
{
"axes": [
1
],
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Max_965"
],
"name": "Broadcast_966",
"op": "Broadcast",
"outputs": [
"Broadcast_966_0"
],
"shape": [
32,
10
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Add_950",
"Broadcast_966"
],
"name": "Subtract_967",
"op": "Subtract",
"outputs": [
"Subtract_967_0"
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Subtract_967"
],
"name": "Exp_968",
"op": "Exp",
"outputs": [
"Exp_968_0"
]
},
{
"element_type": "int64_t",
"name": "Constant_969",
"op": "Constant",
"outputs": [
"Constant_969_0"
],
"shape": [
1
],
"value": [
"1"
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Exp_968",
"Constant_969"
],
"name": "Sum_970",
"op": "Sum",
"outputs": [
"Sum_970_0"
],
"reduction_axes": [
1
]
},
{
"axes": [
1
],
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Sum_970"
],
"name": "Broadcast_971",
"op": "Broadcast",
"outputs": [
"Broadcast_971_0"
],
"shape": [
32,
10
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Broadcast_971"
],
"name": "Log_975",
"op": "Log",
"outputs": [
"Log_975_0"
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Log_975",
"Subtract_967"
],
"name": "Subtract_976",
"op": "Subtract",
"outputs": [
"Subtract_976_0"
]
},
{
"friendly_name": "loss/dense_1_loss/Reshape",
"input_order": [
0,
1
],
"inputs": [
"Parameter_947"
],
"name": "Reshape_962",
"op": "Reshape",
"output_shape": [
32
],
"outputs": [
"Reshape_962_0"
]
},
{
"friendly_name": "loss/dense_1_loss/Cast",
"inputs": [
"Reshape_962"
],
"name": "Convert_963",
"op": "Convert",
"outputs": [
"Convert_963_0"
],
"target_type": "int64_t"
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Convert_963"
],
"name": "OneHot_973",
"one_hot_axis": 1,
"op": "OneHot",
"outputs": [
"OneHot_973_0"
],
"shape": [
32,
10
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"OneHot_973"
],
"name": "Convert_974",
"op": "Convert",
"outputs": [
"Convert_974_0"
],
"target_type": "float"
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Subtract_976",
"Convert_974"
],
"name": "Multiply_977",
"op": "Multiply",
"outputs": [
"Multiply_977_0"
]
},
{
"element_type": "int64_t",
"name": "Constant_978",
"op": "Constant",
"outputs": [
"Constant_978_0"
],
"shape": [
1
],
"value": [
"1"
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Multiply_977",
"Constant_978"
],
"name": "Sum_979",
"op": "Sum",
"outputs": [
"Sum_979_0"
],
"reduction_axes": [
1
]
},
{
"element_type": "int64_t",
"name": "Constant_983",
"op": "Constant",
"outputs": [
"Constant_983_0"
],
"shape": [
1
],
"value": [
"0"
]
},
{
"friendly_name": "loss/dense_1_loss/Sum",
"inputs": [
"Sum_979",
"Constant_983"
],
"name": "Sum_984",
"op": "Sum",
"outputs": [
"Sum_984_0"
],
"reduction_axes": [
0
]
},
{
"inputs": [
"Sum_984"
],
"name": "Result_986",
"needs_default_layout": true,
"op": "Result",
"outputs": [
"Result_986_0"
]
},
{
"element_type": "int32_t",
"friendly_name": "loss/dense_1_loss/num_elements",
"name": "Constant_981",
"op": "Constant",
"outputs": [
"Constant_981_0"
],
"shape": [],
"value": [
"32"
]
},
{
"friendly_name": "loss/dense_1_loss/num_elements/Cast",
"inputs": [
"Constant_981"
],
"name": "Convert_982",
"op": "Convert",
"outputs": [
"Convert_982_0"
],
"target_type": "float"
},
{
"inputs": [
"Convert_982"
],
"name": "Result_987",
"needs_default_layout": true,
"op": "Result",
"outputs": [
"Result_987_0"
]
},
{
"element_type": "int32_t",
"friendly_name": "Adam/gradients/loss/dense_1_loss/Sum_grad/Shape",
"name": "Constant_985",
"op": "Constant",
"outputs": [
"Constant_985_0"
],
"shape": [
1
],
"value": [
"32"
]
},
{
"inputs": [
"Constant_985"
],
"name": "Result_988",
"needs_default_layout": true,
"op": "Result",
"outputs": [
"Result_988_0"
]
},
{
"friendly_name": "dense_1/Softmax",
"inputs": [
"Add_950"
],
"name": "Softmax_951",
"op": "Softmax",
"outputs": [
"Softmax_951_0"
],
"softmax_axes": [
1
]
},
{
"axis": 1,
"friendly_name": "metrics/acc/ArgMax",
"index_element_type": "int64_t",
"inputs": [
"Softmax_951"
],
"name": "ArgMax_952",
"op": "ArgMax",
"outputs": [
"ArgMax_952_0"
]
},
{
"friendly_name": "metrics/acc/Cast",
"inputs": [
"ArgMax_952"
],
"name": "Convert_953",
"op": "Convert",
"outputs": [
"Convert_953_0"
],
"target_type": "float"
},
{
"friendly_name": "metrics/acc/Squeeze",
"input_order": [
0,
1
],
"inputs": [
"Parameter_947"
],
"name": "Reshape_954",
"op": "Reshape",
"output_shape": [
32
],
"outputs": [
"Reshape_954_0"
]
},
{
"friendly_name": "metrics/acc/Equal",
"inputs": [
"Convert_953",
"Reshape_954"
],
"name": "Equal_955",
"op": "Equal",
"outputs": [
"Equal_955_0"
]
},
{
"friendly_name": "metrics/acc/Cast_1",
"inputs": [
"Equal_955"
],
"name": "Convert_956",
"op": "Convert",
"outputs": [
"Convert_956_0"
],
"target_type": "float"
},
{
"element_type": "int64_t",
"name": "Constant_957",
"op": "Constant",
"outputs": [
"Constant_957_0"
],
"shape": [
1
],
"value": [
"0"
]
},
{
"friendly_name": "metrics/acc/Sum",
"inputs": [
"Convert_956",
"Constant_957"
],
"name": "Sum_958",
"op": "Sum",
"outputs": [
"Sum_958_0"
],
"reduction_axes": [
0
]
},
{
"inputs": [
"Sum_958"
],
"name": "Result_989",
"needs_default_layout": true,
"op": "Result",
"outputs": [
"Result_989_0"
]
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Exp_968",
"Broadcast_971"
],
"name": "Divide_972",
"op": "Divide",
"outputs": [
"Divide_972_0"
],
"pythondiv": true
},
{
"friendly_name": "loss/dense_1_loss/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits",
"inputs": [
"Divide_972",
"Convert_974"
],
"name": "Subtract_980",
"op": "Subtract",
"outputs": [
"Subtract_980_0"
]
},
{
"inputs": [
"Subtract_980"
],
"name": "Result_990",
"needs_default_layout": true,
"op": "Result",
"outputs": [
"Result_990_0"
]
},
{
"element_type": "int32_t",
"friendly_name": "metrics/acc/Size",
"name": "Constant_959",
"op": "Constant",
"outputs": [
"Constant_959_0"
],
"shape": [],
"value": [
"32"
]
},
{
"friendly_name": "metrics/acc/Cast_2",
"inputs": [
"Constant_959"
],
"name": "Convert_960",
"op": "Convert",
"outputs": [
"Convert_960_0"
],
"target_type": "float"
},
{
"inputs": [
"Convert_960"
],
"name": "Result_991",
"needs_default_layout": true,
"op": "Result",
"outputs": [
"Result_991_0"
]
}
],
"parameters": [
"Parameter_942",
"Parameter_943",
"Parameter_944",
"Parameter_945",
"Parameter_946",
"Parameter_947"
],
"result": [
"Result_986",
"Result_987",
"Result_988",
"Result_989",
"Result_990",
"Result_991"
]
}
]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment