Home
last modified time | relevance | path

Searched refs:op_device (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_device_copy_conversion.cc63 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()
Dreplicate_invariant_op_hoisting.cc122 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()
Dcross_host_transfer.cc87 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/
Dtensor_handle.h58 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);
Dtensor_handle.cc226 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 …]
Dplacement_utils.cc144 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()
Dexecute.cc105 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/
Dremote_mgr.cc137 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()
Deager_service_impl.cc719 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()
Dremote_copy_node.cc324 local_handle->set_device(h->op_device() ? h->op_device()->name() in SerializePackedHandle()
Deager_service_impl_test.cc1174 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/
Dvirtual_gpu_test.py104 op_device = ()
109 op_device += (device_index,)
111 result.append(op_device)
/external/tensorflow/tensorflow/c/
Dc_test_util.cc210 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()
Dc_test_util.h93 const string& op_device, TF_Status* s,
/external/tensorflow/tensorflow/core/protobuf/
Dremote_tensor_handle.proto29 string op_device = 4; field