corss

[bathiya17]

Description

Please, describe briefly what the issue is

Code example

If possible, write a minimum working example that reproduces the bug,
e.g:

import vegasflow
vegasflow.broken_function()

Additional information

Does the problem occur in CPU or GPU?
If GPU, how many? Which version of Cuda do you have?

nvcc --version

Please include the version of python, vegasflow and tensorflow that you are running.
Running the following python script will produce useful information:

import tensorflow as tf
import sys
from tensorflow.python.framework import test_util
import vegasflow

print(f"Python version: {sys.version}")
print(f"Vegasflow: {vegasflow.__version__}")
print(f"Tensorflow: {tf.__version__}")
print(f"tf-mkl: {test_util.IsMklEnabled()}")
print(f"tf-cuda: {tf.test.is_built_with_cuda()}")
print(f"tf-cuda: {tf.test.is_built_with_rocm()}")
print(f"GPU available: {tf.test.is_gpu_available()}")

[scarlehoff]

?