Home
last modified time | relevance | path

Searched refs:log_device_placement (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dplacer.cc109 void LogDeviceAssignment(const Node* node, bool log_device_placement) { in LogDeviceAssignment() argument
111 if (log_device_placement) { in LogDeviceAssignment()
122 bool log_device_placement) { in AssignAndLog() argument
128 LogDeviceAssignment(node, log_device_placement); in AssignAndLog()
137 bool allow_soft_placement, bool log_device_placement) in Placer() argument
144 log_device_placement_(log_device_placement) {} in Placer()
Dinspecting_placer.cc118 bool log_device_placement) in InspectingPlacer() argument
124 log_device_placement_(log_device_placement) {} in InspectingPlacer()
Dcolocation_graph.h61 bool log_device_placement);
221 bool log_device_placement);
Dinspecting_placer.h75 bool allow_soft_placement, bool log_device_placement);
Dplacer.h74 bool log_device_placement);
Dcolocation_graph.cc250 bool log_device_placement) { in EnsureCompatibilityAcrossResourceEdge() argument
282 if (log_device_placement) { in EnsureCompatibilityAcrossResourceEdge()
609 bool log_device_placement) in ColocationGraph() argument
614 allow_soft_placement, log_device_placement), in ColocationGraph()
622 log_device_placement_(log_device_placement) { in ColocationGraph()
Dplacer_test.cc261 bool allow_soft_placement, bool log_device_placement) { in Place() argument
263 allow_soft_placement, log_device_placement); in Place()
269 bool log_device_placement) { in CallOptPassesAndPlace() argument
300 allow_soft_placement, log_device_placement); in CallOptPassesAndPlace()
309 bool log_device_placement) { in Place() argument
311 log_device_placement); in Place()
319 bool log_device_placement) { in CallOptPassesAndPlace() argument
321 log_device_placement); in CallOptPassesAndPlace()
Dgraph_execution_state.cc625 session_options_->config.log_device_placement()); in InitBaseGraph()
Dprocess_function_library_runtime.cc808 options.config_proto.log_device_placement()); in InstantiateMultiDevice()
Ddirect_session.cc352 if (options.config.log_device_placement()) { in DirectSession()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h105 virtual Status Init(const bool log_device_placement, const NodeDef& ndef,
205 Status Init(const bool log_device_placement, const NodeDef& ndef,
291 Status InstantiateFunc(const bool log_device_placement, const NodeDef& ndef,
294 Status Init(const bool log_device_placement, const NodeDef& ndef,
Dkernel_and_device.cc100 Status KernelAndDeviceOp::Init(const bool log_device_placement, in Init() argument
133 Status KernelAndDeviceFunc::InstantiateFunc(const bool log_device_placement, in InstantiateFunc() argument
216 options.config_proto.set_log_device_placement(log_device_placement); in InstantiateFunc()
223 Status KernelAndDeviceFunc::Init(const bool log_device_placement, in Init() argument
227 InstantiateFunc(log_device_placement, ndef, graph_collector)); in Init()
Dcontext.cc89 log_device_placement_(opts.config.log_device_placement()), in EagerContext()
/external/tensorflow/tensorflow/python/eager/
Ddevice_placement_test.py45 context.context().log_device_placement = True
152 context.context().log_device_placement = True
190 context.context().log_device_placement = True
Dcontext.py961 config.log_device_placement = self._log_device_placement
1657 def log_device_placement(self): member in Context
1658 return self.config.log_device_placement
1660 @log_device_placement.setter
1661 def log_device_placement(self, enable): member in Context
2122 return context().log_device_placement
2132 context().log_device_placement = enabled
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-config-proto.pbtxt63 name: "log_device_placement"
/external/tensorflow/tensorflow/python/ops/
Dinit_ops_test.py195 allow_soft_placement=False, log_device_placement=True)
Dcontrol_flow_ops_test.py1192 allow_soft_placement=False, log_device_placement=True)
/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py200 context.context().log_device_placement)
206 context.context().log_device_placement)
/external/tensorflow/tensorflow/compiler/tests/
Deager_test.py804 config=config_pb2.ConfigProto(log_device_placement=True))
/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py1152 log_device_placement=True,
Dcond_v2_test.py1604 context.context().log_device_placement = True
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto461 bool log_device_placement = 8; field
/external/tensorflow/tensorflow/python/client/
Dsession_test.py1951 config_pb = config_pb2.ConfigProto(log_device_placement=True)