Home
last modified time | relevance | path

Searched refs:SingleMachine (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc41 SingleMachine::SingleMachine(int timeout_s, int num_cpu_cores, int num_gpus) in SingleMachine() function in tensorflow::grappler::SingleMachine
63 SingleMachine::~SingleMachine() { in ~SingleMachine()
71 Status SingleMachine::Provision() { in Provision()
124 Status SingleMachine::Initialize(const GrapplerItem& item) { in Initialize()
136 Status SingleMachine::Shutdown() { in Shutdown()
146 Status SingleMachine::Run(const GraphDef& graph_def, in Run()
207 Status SingleMachine::EnablePeakMemoryStats() { in EnablePeakMemoryStats()
214 Status SingleMachine::GetPeakMemoryUsage( in GetPeakMemoryUsage()
242 Status SingleMachine::RunWithTimeout( in RunWithTimeout()
248 Status SingleMachine::RunWithTimeout( in RunWithTimeout()
[all …]
Dsingle_machine.h31 class SingleMachine : public Cluster {
33 SingleMachine(int timeout_s, int num_cpu_cores, int num_gpus);
34 ~SingleMachine() override;
Dsingle_machine_test.cc52 new SingleMachine(timeout_s, 3 /* num_cpu_cores */, 0 /* num_gpus */)); in SetUp()
65 std::unique_ptr<SingleMachine> cluster_;
331 SingleMachine cluster(5, 3, 0); in RunInfiniteTFLoop()
622 SingleMachine cluster(60 /* timeout_s */, 3 /* num_cpu_cores */, in TEST_F()
/external/tensorflow/tensorflow/c/experimental/grappler/
Dgrappler_test.cc186 std::unique_ptr<SingleMachine> cluster(new SingleMachine(5 * 60, 3, 0)); in TEST()
234 std::unique_ptr<SingleMachine> cluster(new SingleMachine(5 * 60, 3, 0)); in TEST()
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt339 tensorflow::grappler::SingleMachine::SingleMachine
/external/tensorflow/tensorflow/python/grappler/
Dcluster_wrapper.cc87 std::make_unique<tensorflow::grappler::SingleMachine>( in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtrt_convert_api.cc56 *cluster = new grappler::SingleMachine(timeout_s, num_cpu_cores, num_gpus); in NewCluster()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc169 std::make_unique<SingleMachine>(/*timeout_s=*/10, 1, 1); in SetUp()
Dauto_mixed_precision_test.cc110 virtual_cluster_.reset(new SingleMachine(/* timeout_s = */ 10, 1, 1)); in SetUp()
1246 virtual_cluster_.reset(new SingleMachine(/* timeout_s = */ 10, 1, 0)); in SetUp()
1453 virtual_cluster_.reset(new SingleMachine(/* timeout_s = */ 10, 1, 0)); in SetUp()
Dgeneric_layout_optimizer_transposer_test.cc360 std::make_unique<SingleMachine>(/*timeout_s=*/10, 1, 1); in SetUp()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc64 cluster_.reset(new SingleMachine(5 * 60, 3, 0)); in SetUp()
149 std::unique_ptr<SingleMachine> cluster_;