Home
last modified time | relevance | path

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

12

/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()
132 bool log_device_placement) { in AssignAndLog() argument
138 LogDeviceAssignment(node, log_device_placement); in AssignAndLog()
147 bool allow_soft_placement, bool log_device_placement) in Placer() argument
154 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);
223 bool log_device_placement);
Dinspecting_placer.h75 bool allow_soft_placement, bool log_device_placement);
Dplacer.h74 bool log_device_placement);
Dcolocation_graph.cc271 bool log_device_placement) { in EnsureCompatibilityAcrossResourceEdge() argument
305 if (log_device_placement) { in EnsureCompatibilityAcrossResourceEdge()
632 bool log_device_placement) in ColocationGraph() argument
636 allow_soft_placement, log_device_placement), in ColocationGraph()
644 log_device_placement_(log_device_placement) { in ColocationGraph()
Dplacer_test.cc273 bool allow_soft_placement, bool log_device_placement) { in Place() argument
275 allow_soft_placement, log_device_placement); in Place()
281 bool log_device_placement) { in CallOptPassesAndPlace() argument
312 allow_soft_placement, log_device_placement); in CallOptPassesAndPlace()
321 bool log_device_placement) { in Place() argument
323 log_device_placement); in Place()
331 bool log_device_placement) { in CallOptPassesAndPlace() argument
333 log_device_placement); in CallOptPassesAndPlace()
Dgraph_execution_state.cc628 session_options_->config.log_device_placement()); in InitBaseGraph()
Ddirect_session.cc361 if (options.config.log_device_placement()) { in DirectSession()
Dprocess_function_library_runtime.cc923 options.config_proto.log_device_placement()); in InstantiateMultiDevice()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h111 virtual Status Init(const bool log_device_placement, const NodeDef& ndef,
213 Status Init(const bool log_device_placement, const NodeDef& ndef,
312 Status InstantiateFunc(const bool log_device_placement, const NodeDef& ndef,
315 Status Init(const bool log_device_placement, const NodeDef& ndef,
Dkernel_and_device.cc101 Status KernelAndDeviceOp::Init(const bool log_device_placement, in Init() argument
139 Status KernelAndDeviceFunc::InstantiateFunc(const bool log_device_placement, in InstantiateFunc() argument
231 options.config_proto.set_log_device_placement(log_device_placement); in InstantiateFunc()
244 Status KernelAndDeviceFunc::Init(const bool log_device_placement, in Init() argument
248 InstantiateFunc(log_device_placement, ndef, graph_collector)); in Init()
Dcontext.cc162 log_device_placement_(opts.config.log_device_placement()), in EagerContext()
/external/tensorflow/tensorflow/python/eager/
Ddevice_placement_test.py41 context.context().log_device_placement = True
150 context.context().log_device_placement = True
222 context.context().log_device_placement = True
Dcontext.py1086 config.log_device_placement = self._log_device_placement
1846 def log_device_placement(self): member in Context
1847 return self.config.log_device_placement
1849 @log_device_placement.setter
1850 def log_device_placement(self, enable): member in Context
2372 return context().log_device_placement
2407 context().log_device_placement = enabled
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_compat_request_state.h132 bool log_device_placement() const { return log_device_placement_; } in log_device_placement() function
Dkernel_fallback_execute_compat.cc503 if (fallback_request_state.log_device_placement() || VLOG_IS_ON(1)) { in KernelFallbackExecuteOpInternal()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-config-proto.pbtxt63 name: "log_device_placement"
/external/tensorflow/tensorflow/python/ops/
Dinit_ops_test.py190 allow_soft_placement=False, log_device_placement=True)
Dcontrol_flow_ops_test.py1195 allow_soft_placement=False, log_device_placement=True)
/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py194 context.context().log_device_placement)
199 context.context().log_device_placement)
/external/tensorflow/tensorflow/compiler/tests/
Deager_test.py801 config=config_pb2.ConfigProto(log_device_placement=True))
/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
Dfunctional_ops_test.py1221 log_device_placement=True,
Dcond_v2_test.py1605 context.context().log_device_placement = True
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto520 bool log_device_placement = 8; field

12