Searched refs:StagingArea (Results 1 – 4 of 4) sorted by relevance
39 stager = data_flow_ops.StagingArea([dtypes.float32])59 stager = data_flow_ops.StagingArea([dtypes.float32, dtypes.float32])80 stager = data_flow_ops.StagingArea(107 stager = data_flow_ops.StagingArea([dtypes.float32])124 stager = data_flow_ops.StagingArea(148 stager = data_flow_ops.StagingArea(176 stager = data_flow_ops.StagingArea(240 stager = data_flow_ops.StagingArea(
20 from tensorflow.python.ops.data_flow_ops import StagingArea
1703 class StagingArea(BaseStagingArea): class1778 super(StagingArea, self).__init__(dtypes, shapes, names, shared_name,
1348 * Added colocation constraints to `StagingArea`.1546 * Add new queue-like `StagingArea` and new ops: `stage` and `unstage`.