Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_device_propagation.cc88 llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { in PopulateDeviceForOpResults() argument
99 value_to_device.insert({result, device}); in PopulateDeviceForOpResults()
116 const llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { in FindDeviceFromOperands() argument
127 auto it = value_to_device.find(operand); in FindDeviceFromOperands()
128 if (it == value_to_device.end()) return llvm::StringRef(); in FindDeviceFromOperands()
144 llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { in PropagateDevicesFromArguments() argument
151 value_to_device.insert({arg, arg_device_attr.getValue()}); in PropagateDevicesFromArguments()
160 llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { in PropagateDevicesInGraph() argument
182 value_to_device); in PropagateDevicesInGraph()
192 FindDeviceFromOperands(*op_to_update, value_to_device); in PropagateDevicesInGraph()
[all …]