{"record":{"id":"6495f50019eb1901","repo":"tensorflow/models","slug":"steps-should-be-an-tensor-python-object-may-cause-6495f5","errorCode":null,"errorMessage":"steps should be an Tensor. Python object may cause retracing.","messagePattern":"steps should be an Tensor\\. Python object may cause retracing\\.","errorType":"exception","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"official/legacy/xlnet/training_utils.py","lineNumber":202,"sourceCode":"      if input_meta_data[\"mem_len\"] > 0:\n        return mem\n\n    def train_steps(iterator, steps):\n      \"\"\"Performs distributed training steps in a loop.\n\n      Args:\n        iterator: the distributed iterator of training datasets.\n        steps: an tf.int32 integer tensor to specify number of steps to run\n          inside host training loop.\n\n      Raises:\n        ValueError: Any of the arguments or tensor shapes are invalid.\n\n      Returns:\n        logits: logits computed.\n      \"\"\"\n      if not isinstance(steps, tf.Tensor):\n        raise ValueError(\"steps should be an Tensor. Python object may cause \"\n                         \"retracing.\")\n\n      def cache_fn():\n        \"\"\"Initializes memory tensor used in XLNet pretraining.\"\"\"\n        mems = []\n        if input_meta_data[\"mem_len\"] > 0:\n          for _ in range(input_meta_data[\"n_layer\"]):\n            zeros = tf.zeros([\n                input_meta_data[\"batch_size_per_core\"],\n                input_meta_data[\"mem_len\"],\n                input_meta_data[\"d_model\"]\n            ],\n                             dtype=tf.float32)\n            mems.append(zeros)\n        return mems\n\n      if input_meta_data[\"mem_len\"] > 0:\n        mem = strategy.run(cache_fn)","sourceCodeStart":184,"sourceCodeEnd":220,"githubUrl":"https://github.com/tensorflow/models/blob/e006f5f0d534913e49c1f1dae87364039fa607e2/official/legacy/xlnet/training_utils.py#L184-L220","documentation":"Error \"steps should be an Tensor. Python object may cause retracing.\" thrown in tensorflow/models.","triggerScenarios":"Thrown at official/legacy/xlnet/training_utils.py:202 when the library encounters an invalid state.","commonSituations":"See trigger scenarios.","solutions":[],"exampleFix":null,"handlingStrategy":null,"validationCode":null,"typeGuard":null,"tryCatchPattern":null,"preventionTips":[],"tags":[],"backgroundTag":null,"analyzedSha":"e006f5f0d534913e49c1f1dae87364039fa607e2","analyzedAt":"2026-08-24T14:09:15.576Z","schemaVersion":2},"datasetVersion":"2026-08-24T17:17:21.512Z"}