B )`k @sdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddlm Z ddl m Z dd l m Z dd l mZd d d ZdS)z,Utility ops shared across tf.contrib.signal.)absolute_import)division)print_functionN)ops) tensor_util) array_ops)control_flow_ops)math_opsc Cst|d||gt|}t|}|jd|jd|jjsVtd|j|jjsltd|jt |}t |}|dk r|dk rt j j dkrt j}ntj}t|||Sdd}d d}tj||||gd d \}}|SQRXdS) azReturns the greatest common divisor via Euclid's algorithm. Args: a: The dividend. A scalar integer `Tensor`. b: The divisor. A scalar integer `Tensor`. name: An optional name for the operation. Returns: A scalar `Tensor` representing the greatest common divisor between `a` and `b`. Raises: ValueError: If `a` or `b` are not scalar integers. gcdrz"a must be an integer type. Got: %sz"b must be an integer type. Got: %sNcSst|t|S)N)r ZgreaterrZ zeros_like)_brT/home/dcms/DCMS/lib/python3.7/site-packages/tensorflow/python/ops/signal/util_ops.pyFzgcd..cSs|t||gS)N)r mod)ar rrrrGrF)Z back_prop)rZ name_scopeZconvert_to_tensorshapeZassert_has_rankZdtype is_integer ValueErrorrZconstant_valuesys version_infomajor fractionsr mathrZ while_loop)rr nameZconst_aZconst_bZmath_gcdZcondbodyrrrr s(       r )N)__doc__ __future__rrrrrrZtensorflow.python.frameworkrrZtensorflow.python.opsrrr r rrrrs