W3cubDocs

/TensorFlow 2.4

Module: tf.compat.v1.test

Testing.

Classes

class Benchmark: Abstract class that provides helpers for TensorFlow benchmarks.

class StubOutForTesting: Support class for stubbing methods out for unit testing.

class TestCase: Base class for tests that need to test TensorFlow.

Functions

assert_equal_graph_def(...): Asserts that two GraphDefs are (mostly) the same.

benchmark_config(...): Returns a tf.compat.v1.ConfigProto for disabling the dependency optimizer.

compute_gradient(...): Computes and returns the theoretical and numerical Jacobian. (deprecated)

compute_gradient_error(...): Computes the gradient error. (deprecated)

create_local_cluster(...): Create and start local servers and return the associated Server objects.

get_temp_dir(...): Returns a temporary directory for use during tests.

gpu_device_name(...): Returns the name of a GPU device if available or the empty string.

is_built_with_cuda(...): Returns whether TensorFlow was built with CUDA (GPU) support.

is_built_with_gpu_support(...): Returns whether TensorFlow was built with GPU (i.e. CUDA or ROCm) support.

is_built_with_rocm(...): Returns whether TensorFlow was built with ROCm (GPU) support.

is_built_with_xla(...): Returns whether TensorFlow was built with XLA support.

is_gpu_available(...): Returns whether TensorFlow can access a GPU. (deprecated)

main(...): Runs all unit tests.

test_src_dir_path(...): Creates an absolute test srcdir path given a relative path.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/test