blob: 5c85c0628ec6a7b86b4926a1efea4b1182c9e692 [file] [log] [blame]
path: "tensorflow.test"
tf_module {
member {
name: "Benchmark"
mtype: "<class \'tensorflow.python.platform.benchmark._BenchmarkRegistrar\'>"
}
member {
name: "StubOutForTesting"
mtype: "<type \'type\'>"
}
member {
name: "TestCase"
mtype: "<type \'type\'>"
}
member {
name: "mock"
mtype: "<type \'module\'>"
}
member_method {
name: "assert_equal_graph_def"
argspec: "args=[\'actual\', \'expected\', \'checkpoint_v2\', \'hash_table_shared_name\'], varargs=None, keywords=None, defaults=[\'False\', \'False\'], "
}
member_method {
name: "benchmark_config"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "compute_gradient"
argspec: "args=[\'x\', \'x_shape\', \'y\', \'y_shape\', \'x_init_value\', \'delta\', \'init_targets\', \'extra_feed_dict\'], varargs=None, keywords=None, defaults=[\'None\', \'0.001\', \'None\', \'None\'], "
}
member_method {
name: "compute_gradient_error"
argspec: "args=[\'x\', \'x_shape\', \'y\', \'y_shape\', \'x_init_value\', \'delta\', \'init_targets\', \'extra_feed_dict\'], varargs=None, keywords=None, defaults=[\'None\', \'0.001\', \'None\', \'None\'], "
}
member_method {
name: "create_local_cluster"
argspec: "args=[\'num_workers\', \'num_ps\', \'protocol\', \'worker_config\', \'ps_config\'], varargs=None, keywords=None, defaults=[\'grpc\', \'None\', \'None\'], "
}
member_method {
name: "disable_with_predicate"
argspec: "args=[\'pred\', \'skip_message\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "get_temp_dir"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "gpu_device_name"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "is_built_with_cuda"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "is_built_with_gpu_support"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "is_built_with_rocm"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "is_built_with_xla"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "is_gpu_available"
argspec: "args=[\'cuda_only\', \'min_cuda_compute_capability\'], varargs=None, keywords=None, defaults=[\'False\', \'None\'], "
}
member_method {
name: "main"
argspec: "args=[\'argv\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "test_src_dir_path"
argspec: "args=[\'relative_path\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "with_eager_op_as_function"
argspec: "args=[\'cls\', \'only_as_function\'], varargs=None, keywords=None, defaults=[\'None\', \'False\'], "
}
}