Searched refs:StagingArea (Results 1 – 3 of 3) 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(
1819 class StagingArea(BaseStagingArea): class1894 super(StagingArea, self).__init__(dtypes, shapes, names, shared_name,
4021 * Added colocation constraints to `StagingArea`.4219 * Add new queue-like `StagingArea` and new ops: `stage` and `unstage`.