Searched refs:value_to_device (Results 1 – 1 of 1) sorted by relevance
88 llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { in PopulateDeviceForOpResults() argument99 value_to_device.insert({result, device}); in PopulateDeviceForOpResults()116 const llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { in FindDeviceFromOperands() argument127 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() argument151 value_to_device.insert({arg, arg_device_attr.getValue()}); in PropagateDevicesFromArguments()160 llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { in PropagateDevicesInGraph() argument182 value_to_device); in PropagateDevicesInGraph()192 FindDeviceFromOperands(*op_to_update, value_to_device); in PropagateDevicesInGraph()[all …]