TPUExecuteAndUpdateVariables
Stay organized with collections
Save and categorize content based on your preferences.
Op that executes a program with optional in-place variable updates.
It (optionally) reads device variables, loads and executes a TPU program on a
TPU device, and then (optionally) in-place updates variables using the program
outputs, as specified in attributes device_var_reads_indices (program input
indices from directly reading variables) and device_var_updates_indices (program
output indices used to update variables, -1 means no-update/read-only). Such
program outputs are consumed by these variables will not appear in the op
output. For the internal use of the distributed TPU compiler.
Public Methods
static
TPUExecuteAndUpdateVariables
|
create
(
Scope
scope, Iterable<
Operand
<?>> args,
Operand
<String> key, List<Class<?>> Tresults, List<Long> deviceVarReadsIndices, List<Long> deviceVarUpdatesIndices)
Factory method to create a class wrapping a new TPUExecuteAndUpdateVariables operation.
|
Iterator<
Operand
<Object>>
|
|
List<
Output
<?>>
|
|
Inherited Methods
From class
java.lang.Object
boolean
|
equals
(Object arg0)
|
final
Class<?>
|
getClass
()
|
int
|
hashCode
()
|
final
void
|
notify
()
|
final
void
|
notifyAll
()
|
String
|
toString
()
|
final
void
|
wait
(long arg0, int arg1)
|
final
void
|
wait
(long arg0)
|
final
void
|
wait
()
|
From interface
java.lang.Iterable
abstract
void
|
forEach
(Consumer<? super T> arg0)
|
abstract
Iterator<
Operand
<Object>>
|
iterator
()
|
abstract
Spliterator<
Operand
<Object>>
|
spliterator
()
|
Public Methods
public
static
TPUExecuteAndUpdateVariables
create
(
Scope
scope, Iterable<
Operand
<?>> args,
Operand
<String> key, List<Class<?>> Tresults, List<Long> deviceVarReadsIndices, List<Long> deviceVarUpdatesIndices)
Factory method to create a class wrapping a new TPUExecuteAndUpdateVariables operation.
Returns
-
a new instance of TPUExecuteAndUpdateVariables
public
Iterator<
Operand
<Object>>
iterator
()
public
List<
Output
<?>>
results
()
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2021-08-16 UTC.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Missing the information I need"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Too complicated / too many steps"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Out of date"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Samples / code issue"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Other"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Easy to understand"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Solved my problem"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Other"
}]
{"lastModified": "Last updated 2021-08-16 UTC."}