Searched refs:allow_soft_placement_ (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | inspecting_placer.cc | 123 allow_soft_placement_(allow_soft_placement), in InspectingPlacer()
150 default_device_, allow_soft_placement_, log_device_placement_); in ComputeIOColocationGroups()
|
| D | inspecting_placer.h | 87 const bool allow_soft_placement_; variable
|
| D | placer.h | 101 const bool allow_soft_placement_; variable
|
| D | placer.cc | 143 allow_soft_placement_(allow_soft_placement), in Placer()
176 default_local_device_, allow_soft_placement_, in Run()
|
| D | colocation_graph.cc | 621 allow_soft_placement_(allow_soft_placement), in ColocationGraph()
1017 if (!allow_soft_placement_) { in ColocateNodeToGroup()
1065 allow_soft_placement_); in ColocateNodes()
1161 return root_member.LimitToPossibleDevices(devices, allow_soft_placement_); in LimitToPossibleDevices()
1200 if (devices.empty() && allow_soft_placement_) { in GetDevicesForNode()
|
| D | colocation_graph.h | 378 const bool allow_soft_placement_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 265 bool AllowSoftPlacement() const { return allow_soft_placement_; } in AllowSoftPlacement()
267 allow_soft_placement_ = enable; in SetAllowSoftPlacement()
636 std::atomic<bool> allow_soft_placement_; variable
|
| D | context.cc | 90 allow_soft_placement_(opts.config.allow_soft_placement()), in EagerContext()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 189 bool allow_soft_placement_; member in tensorflow::tensorrt::TRTEngineOp
365 status = context->GetAttr("_allow_soft_placement", &allow_soft_placement_); in TRTEngineOp()
367 allow_soft_placement_ = true; in TRTEngineOp()
455 allow_soft_placement_, ctx->num_inputs(), in ExecuteNativeSegment()
|