Модуль: tf.test

Публичный API пространства имён tf._api.v2.test

Модули

experimental модуль: публичный API пространства имён tf._api.v2.test.experimental

Классы

class Benchmark: абстрактный класс, предоставляющий вспомогательные средства для тестирования производительности TensorFlow.

class TestCase: базовый класс для тестов, предназначенных для проверки TensorFlow.

Функции

assert_equal_graph_def(...): проверяет, что два объекта GraphDef в основном одинаковы.

benchmark_config(...): возвращает tf.compat.v1.ConfigProto для отключения оптимизатора зависимостей.

compute_gradient(...): вычисляет теоретическую и численную матрицы Якоби для f.

create_local_cluster(...): создаёт и запускает локальные серверы и возвращает связанные с ними объекты Server.

disable_with_predicate(...): отключает тест, если pred имеет значение true.

gpu_device_name(...): возвращает имя устройства GPU, если оно доступно, или пустую строку.

is_built_with_cuda(...): возвращает, была ли TensorFlow собрана с поддержкой CUDA (GPU).

is_built_with_gpu_support(...): возвращает, была ли TensorFlow собрана с поддержкой GPU (CUDA или ROCm).

is_built_with_rocm(...): возвращает, была ли TensorFlow собрана с поддержкой ROCm (GPU).

is_built_with_xla(...): возвращает, была ли TensorFlow собрана с поддержкой XLA.

is_gpu_available(...): возвращает, может ли TensorFlow получить доступ к GPU. (устарело)

main(...): запускает все модульные тесты.

with_eager_op_as_function(...): возвращает тот же класс. Этот элемент будет удалён, когда его использование будет полностью прекращено.