tf.raw_ops.MapStage
Stage (key, values) in the underlying container which behaves like a hashtable.
tf.raw_ops.MapStage( key, indices, values, dtypes, capacity=0, memory_limit=0, container='', shared_name='', name=None )
Args | |
---|---|
key |
A Tensor of type int64 . int64 |
indices |
A Tensor of type int32 . |
values |
A list of Tensor objects. a list of tensors dtypes A list of data types that inserted values should adhere to. |
dtypes |
A list of tf.DTypes . |
capacity |
An optional int that is >= 0 . Defaults to 0 . Maximum number of elements in the Staging Area. If > 0, inserts on the container will block when the capacity is reached. |
memory_limit |
An optional int that is >= 0 . Defaults to 0 . |
container |
An optional string . Defaults to "" . If non-empty, this queue is placed in the given container. Otherwise, a default container is used. |
shared_name |
An optional string . Defaults to "" . It is necessary to match this name to the matching Unstage Op. |
name |
A name for the operation (optional). |
Returns | |
---|---|
The created Operation. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/raw_ops/MapStage