Searched refs:SingleMachine (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 41 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()
122 Status SingleMachine::Initialize(const GrapplerItem& item) { in Initialize()
134 Status SingleMachine::Shutdown() { in Shutdown()
144 Status SingleMachine::Run(const GraphDef& graph_def, in Run()
208 Status SingleMachine::EnablePeakMemoryStats(bool enable) { in EnablePeakMemoryStats()
215 Status SingleMachine::GetPeakMemoryUsage( in GetPeakMemoryUsage()
243 Status SingleMachine::RunWithTimeout( in RunWithTimeout()
249 Status SingleMachine::RunWithTimeout( in RunWithTimeout()
[all …]
|
| D | single_machine.h | 31 class SingleMachine : public Cluster {
33 SingleMachine(int timeout_s, int num_cpu_cores, int num_gpus);
34 ~SingleMachine() override;
|
| D | single_machine_test.cc | 47 new SingleMachine(timeout_s, 3 /* num_cpu_cores */, 0 /* num_gpus */)); in SetUp()
60 std::unique_ptr<SingleMachine> cluster_;
318 SingleMachine cluster(5, 3, 0); in RunInfiniteTFLoop()
608 SingleMachine cluster(60 /* timout_s */, 3 /* num_cpu_cores */, in TEST_F()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 139 new tensorflow::grappler::SingleMachine( in TF_NewCluster()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 128 new grappler::SingleMachine( in ConvertGraphDefToTensorRT()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 46 cluster_.reset(new SingleMachine(5 * 60, 3, 0)); in SetUp()
118 std::unique_ptr<SingleMachine> cluster_;
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 44 virtual_cluster_.reset(new SingleMachine(/* timeout_s = */ 10, 1, 1)); in SetUp()
|