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 GraphDef
s 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.
disable_with_predicate(...)
: Disables the test if pred is true.
get_temp_dir(...)
: Returns a temporary directory for use during tests.
gpu_device_name(...)
: Returns the name of a GPU device if available or a 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 (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.
with_eager_op_as_function(...)
: Returns the same class. This will be removed once all usages are removed.