From 2bcac3380994ae0b998d643fd9369c3929a492c6 Mon Sep 17 00:00:00 2001
From: dataangel <dataangel@live.com>
Date: Mon, 14 Dec 2020 05:27:52 +0800
Subject: [PATCH] Update gen_math_ops.cs

---
 src/TensorFlowNET.Core/Operations/gen_math_ops.cs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/TensorFlowNET.Core/Operations/gen_math_ops.cs b/src/TensorFlowNET.Core/Operations/gen_math_ops.cs
index fa74bffe..67b5b614 100644
--- a/src/TensorFlowNET.Core/Operations/gen_math_ops.cs
+++ b/src/TensorFlowNET.Core/Operations/gen_math_ops.cs
@@ -1,4 +1,4 @@
-/*****************************************************************************
+/*****************************************************************************
    Copyright 2018 The TensorFlow.NET Authors. All Rights Reserved.
 
    Licensed under the Apache License, Version 2.0 (the "License");
@@ -619,6 +619,16 @@ namespace Tensorflow
 
         public static Tensor squared_difference(Tensor x, Tensor y, string name = null)
         {
+            if (tf.Context.executing_eagerly())
+            {
+                var results = tf.Runner.TFE_FastPathExecute(tf.Context, tf.Context.DeviceName,
+                    "SquaredDifference", name,
+                    null,
+                    x,y);
+
+                return results[0];
+            }
+
             var _op = tf.OpDefLib._apply_op_helper("SquaredDifference", name, args: new { x, y, name });
 
             return _op.outputs[0];
@@ -1210,4 +1220,4 @@ namespace Tensorflow
             return _op.outputs[0];
         }
     }
-}
\ No newline at end of file
+}