public static class XlaHostCompute.Options extends Object
XlaHostCompute
Modifier and Type | Method and Description |
---|---|
XlaHostCompute.Options |
costEstimateNs(Long costEstimateNs)
Sets the costEstimateNs option.
|
XlaHostCompute.Options |
recvKey(String recvKey)
Sets the recvKey option.
|
XlaHostCompute.Options |
sendKey(String sendKey)
Sets the sendKey option.
|
XlaHostCompute.Options |
tpuCore(Long tpuCore)
Sets the tpuCore option.
|
public XlaHostCompute.Options sendKey(String sendKey)
sendKey
- the sendKey optionpublic XlaHostCompute.Options recvKey(String recvKey)
recvKey
- the recvKey optionpublic XlaHostCompute.Options costEstimateNs(Long costEstimateNs)
costEstimateNs
- Estimated duration of the host computation in nanoseconds.public XlaHostCompute.Options tpuCore(Long tpuCore)
tpuCore
- Default core to use for host to device transfers.Copyright © 2015–2022. All rights reserved.