MCPcopy Create free account
hub / github.com/alibaba/nann / generate_dequeue_op

Method generate_dequeue_op

tensorflow/tensorflow/python/tpu/tpu_feed.py:460–497  ·  view source on GitHub ↗

Generates the device-side Op to dequeue a tuple from the queue. Implicitly freezes the queue configuration if it is not already frozen, which will raise errors if the shapes and types have not been fully specified. Args: tpu_device: The TPU device ordinal where the infeed ins

(self, tpu_device=0)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

body_wrapperFunction · 0.45

Calls 3

freezeMethod · 0.95
get_sharded_shapeMethod · 0.80
deviceMethod · 0.45

Tested by

no test coverage detected