/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | tensor_device_copy_conversion.cc | 63 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnFunction() local 64 if (!op_device) op_device = empty_string; in runOnFunction() 67 if (op_device != arg_device) return WalkResult::advance(); in runOnFunction()
|
D | replicate_invariant_op_hoisting.cc | 122 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in UsesVirtualDevice() local 123 if (!op_device) return WalkResult::advance(); in UsesVirtualDevice() 125 if (virtual_devices.getValue().get(op_device.getValue())) in UsesVirtualDevice()
|
D | cross_host_transfer.cc | 87 std::string op_device = ""; in runOnFunction() local 89 op_device = device_attr.getValue().str(); in runOnFunction() 91 std::string dst_host = GetHost(op_device); in runOnFunction()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | tensor_handle.h | 58 TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device, 61 TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device, 63 TensorHandle(Device* d, Device* op_device, Device* resource_device, 78 Device* op_device, EagerContext* ctx); 80 Device* op_device, 83 static TensorHandle* CreateEmptyLocalHandle(Device* d, Device* op_device, 151 Device* op_device() const { return op_device_; } in op_device() function 205 uint64 context_view_id, string op_device);
|
D | tensor_handle.cc | 226 Device* op_device, in CreateLocalHandle() argument 228 return CreateLocalHandle(std::move(t), d, op_device, nullptr, ctx); in CreateLocalHandle() 232 Device* op_device, in CreateLocalHandle() argument 236 return new TensorHandle(std::move(t), d, op_device, ctx); in CreateLocalHandle() 238 return new TensorHandle(std::move(t), d, op_device, resource_device, ctx); in CreateLocalHandle() 242 TensorHandle::TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device, in TensorHandle() argument 247 op_device_(op_device), in TensorHandle() 258 TensorHandle::TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device, in TensorHandle() argument 263 op_device_(op_device), in TensorHandle() 278 TensorHandle* TensorHandle::CreateEmptyLocalHandle(Device* d, Device* op_device, in CreateEmptyLocalHandle() argument [all …]
|
D | placement_utils.cc | 144 Device* op_device = op.Device() == kVariantDeviceNull in MaybePinToResourceDevice() local 158 << ", op device = " << op_device->name(); in MaybePinToResourceDevice() 164 if (resource_device != op_device || op.Device() == kVariantDeviceNull) { in MaybePinToResourceDevice() 165 DVLOG(1) << (resource_device != op_device ? "Changing " : "Setting ") in MaybePinToResourceDevice()
|
D | execute.cc | 105 Device* op_device, in CopyInputToExpectedDevice() argument 113 const string& op_device_name = DeviceNameOrUnspecified(op_device); in CopyInputToExpectedDevice() 208 if ((h->op_device() != nullptr) && in ValidateInputTypeAndPlacement() 209 (h->op_device()->name() == op->DeviceName())) { in ValidateInputTypeAndPlacement() 847 tensorflow::Device* op_device = absl::get<Device*>(op->Device()); in EagerRemoteExecute() local 860 if (op_device != input_device && in EagerRemoteExecute() 864 !ctx.OnSameTask(op_device, input_device)) { in EagerRemoteExecute() 868 ctx.CPUDeviceOnTask(op_device, &remote_cpu_device)); in EagerRemoteExecute() 879 &ctx, op, op_device, handle, i, handle_device, in EagerRemoteExecute() 891 (!input->HasResourceShapeMirror(op_device, in EagerRemoteExecute() [all …]
|
/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
D | remote_mgr.cc | 137 out->set_op_device(in->op_device() ? in->op_device()->name() : ""); in SerializeRemoteTensorHandle() 157 if (parent_->local_device_mgr()->LookupDevice(in.op_device(), &device).ok() || in DeserializeRemoteTensorHandle() 165 in.op_device().empty() ? in.device() : in.op_device(); in DeserializeRemoteTensorHandle()
|
D | eager_service_impl.cc | 719 Device* op_device = nullptr; in SendPackedHandle() local 721 item.local_handle().device().c_str(), &op_device)); in SendPackedHandle() 723 std::move(tensor), /*d=*/nullptr, op_device, eager_context); in SendPackedHandle()
|
D | remote_copy_node.cc | 324 local_handle->set_device(h->op_device() ? h->op_device()->name() in SerializePackedHandle()
|
D | eager_service_impl_test.cc | 1174 EXPECT_EQ(handle0->op_device()->name(), device0); in TEST_F() 1187 EXPECT_EQ(handle1->op_device()->name(), device1); in TEST_F() 1195 EXPECT_EQ(handle2->op_device()->name(), device2); in TEST_F()
|
/external/tensorflow/tensorflow/python/client/ |
D | virtual_gpu_test.py | 104 op_device = () 109 op_device += (device_index,) 111 result.append(op_device)
|
/external/tensorflow/tensorflow/c/ |
D | c_test_util.cc | 210 TF_Operation** op, const string& op_device, bool check) { in BinaryOpHelper() argument 212 if (!op_device.empty()) { in BinaryOpHelper() 213 TF_SetDevice(desc, op_device.c_str()); in BinaryOpHelper() 225 const string& op_device, TF_Status* s, in MinWithDevice() argument 228 BinaryOpHelper("Min", l, r, graph, s, name, &op, op_device, true); in MinWithDevice()
|
D | c_test_util.h | 93 const string& op_device, TF_Status* s,
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | remote_tensor_handle.proto | 29 string op_device = 4; field
|