Searched defs:host_device (Results 1 – 11 of 11) sorted by relevance
638 std::string host_device; in TEST() local659 std::string host_device; in TEST() local679 std::string host_device; in TEST() local699 std::string host_device; in TEST() local722 std::string host_device; in TEST() local748 std::string host_device; in TEST() local774 std::string host_device; in TEST() local803 std::string host_device; in TEST() local
491 std::string* host_device) { in GetHostDeviceOutsideComputation()
61 void WrapOpInLaunch(Operation* host_op, llvm::StringRef host_device) { in WrapOpInLaunch()89 std::string host_device; in runOnOperation() local
88 llvm::StringRef host_device) { in CreateLaunchForBlock()183 llvm::StringRef host_device) { in CreateHeadComputation()204 std::string* host_device, bool* cluster_updated) { in LiftHeadOutsideCompiledOps()291 llvm::StringRef host_device) { in CreateTailComputation()349 const mlir::TF::RuntimeDevices& devices, std::string host_device, in LiftTailOutsideCompiledOps()434 std::string host_device; in runOnOperation() local
183 OpBuilder& builder, Operation* loc_op, llvm::StringRef host_device) { in CreateLaunchOpForOutsideCluster()424 llvm::StringRef host_device) { in CreateParallelExecuteForOutsideCompilation()499 std::string host_device; in runOnOperation() local
299 const auto host_device = host_devices[i]; in Run() local307 for (const auto host_device : host_devices) { in Run() local325 for (const auto host_device : host_devices) { in Run() local385 for (const auto host_device : host_devices) { in Run() local
7 pub mod host_device; module
192 let host_device = Box::new(HostDevice::new( in handle_attach_device() localVariable
162 def host_device(self, member in DeviceAssignment
87 string host_device = 7; field
687 string host_device() const { return node_.host_device(); } in host_device() function