Searched defs:op_device (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_device_copy_conversion.cc | 52 const StringAttr &op_device) { in runOnOperation()
87 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation() local
96 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation() local
|
| D | replicate_invariant_op_hoisting.cc | 92 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in UsesVirtualDevice() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | remote_tensor_handle.proto | 29 string op_device = 4; field
|
| /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 | 157 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 | 123 Device* op_device, in CopyInputToExpectedDevice()
297 const Device* op_device) { in GetKernelDef()
308 const Device* op_device, const KernelDef* kernel_def, in IsHostMemoryArg()
969 Device* op_device = nullptr; in ExtractFunctionInputInfo() local
1556 tensorflow::Device* op_device = absl::get<Device*>(op->Device()); in EagerRemoteExecute() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| D | conversion_function.cc | 47 Device* op_device, EagerContext* eager_ctx) { in CopyRefGpuTensorToRuntimeFallbackTensor()
|
| /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 | 758 Device* op_device = nullptr; in SendPackedHandle() local
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 2834 StringRef op_device = TFOp(v_op).device(); in matchAndRewrite() local
|