Searched defs:op_device (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | remote_tensor_handle.proto | 29 string op_device = 4; field
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_device_copy_conversion.cc | 63 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnFunction() local
|
| D | cross_host_transfer.cc | 87 std::string op_device = ""; in runOnFunction() local
|
| D | replicate_invariant_op_hoisting.cc | 122 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in UsesVirtualDevice() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 226 Device* op_device, in CreateLocalHandle()
232 Device* op_device, in CreateLocalHandle()
242 TensorHandle::TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device, in TensorHandle()
258 TensorHandle::TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device, in TensorHandle()
278 TensorHandle* TensorHandle::CreateEmptyLocalHandle(Device* d, Device* op_device, in CreateEmptyLocalHandle()
285 TensorHandle::TensorHandle(Device* d, Device* op_device, in TensorHandle()
825 string op_device) { in SetRemoteShapeAndDevice()
|
| D | tensor_handle.h | 151 Device* op_device() const { return op_device_; } in op_device() function
|
| D | placement_utils.cc | 144 Device* op_device = op.Device() == kVariantDeviceNull in MaybePinToResourceDevice() local
|
| D | execute.cc | 105 Device* op_device, in CopyInputToExpectedDevice()
847 tensorflow::Device* op_device = absl::get<Device*>(op->Device()); in EagerRemoteExecute() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.cc | 210 TF_Operation** op, const string& op_device, bool check) { in BinaryOpHelper()
225 const string& op_device, TF_Status* s, in MinWithDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 719 Device* op_device = nullptr; in SendPackedHandle() local
|