Searched refs:kFuncDeviceAttr (Results 1 – 5 of 5) sorted by relevance
39 constexpr char kFuncDeviceAttr[] = "tf.device"; variable144 func.getArgAttrOfType<StringAttr>(arg.getArgNumber(), kFuncDeviceAttr); in PropagateDevicesFromArguments()219 operand.getOperandNumber(), kFuncDeviceAttr); in PropagateDevicesToResults()221 func.setResultAttr(operand.getOperandNumber(), kFuncDeviceAttr, in PropagateDevicesToResults()
54 constexpr char kFuncDeviceAttr[] = "tf.device"; variable131 func.getArgAttrOfType<mlir::StringAttr>(arg_no, kFuncDeviceAttr); in GetDeviceAttr()159 func_op.setArgAttr(arg.getArgNumber(), kFuncDeviceAttr, in ComputeResourceDevicesInComputation()
53 constexpr char kFuncDeviceAttr[] = "tf.device"; variable121 kFuncDeviceAttr); in IsSupportedInputOp()
56 constexpr char kFuncDeviceAttr[] = "tf.device"; variable168 resource_arg.getArgNumber(), kFuncDeviceAttr); in BuildVariableAccessInfo()
61 constexpr char kFuncDeviceAttr[] = "tf.device"; variable