tensorflow

This module tests TensorFlow in available modules containing substring 'TensorFlow'. The test itself is based on an official multi-worker with Keras tutoral at https://www.tensorflow.org/tutorials/distribute/multi_worker_with_keras

EESSI_TensorFlow

Bases: RunOnlyRegressionTest, EESSI_Mixin

assert_completion()

Assert that the test ran until completion

assert_convergence()

Assert that the network learned something during training

assert_sanity()

Check all sanity criteria

assert_tf_config_ranks()

Assert that each rank sets a TF_CONFIG

set_compute_unit()

Set the compute unit to which tasks will be assigned: one task per CPU socket for CPU runs, and one task per GPU for GPU runs.

set_executable_opts()

Set executable opts based on device_type parameter

set_thread_count_args()

Set executable opts defining the thread count