Home
last modified time | relevance | path

Searched refs:num_input_pipelines (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dparameter_server_strategy.py362 num_input_pipelines = multi_worker_util.worker_count(
366 num_input_pipelines = 1
368 num_input_pipelines=num_input_pipelines,
383 num_input_pipelines = multi_worker_util.worker_count(
387 num_input_pipelines = 1
390 num_input_pipelines=num_input_pipelines,
H A Dinput_lib_type_spec_test.py286 dataset = dataset.shard(input_context.num_input_pipelines,
426 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
537 dataset = dataset.shard(input_context.num_input_pipelines,
600 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
661 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
722 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
H A Ddistribute_lib_test.py580 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
583 self.assertEqual(2, input_context.num_input_pipelines)
587 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
594 num_input_pipelines=1, input_pipeline_id=0, num_replicas_in_sync=42)
599 num_input_pipelines=3, input_pipeline_id=1, num_replicas_in_sync=42)
H A Dinput_lib_test.py87 num_input_pipelines=input_workers.num_workers,
1121 return dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
1239 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
1284 return dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
1840 num_input_pipelines=num_workers,
1901 num_input_pipelines=num_workers,
H A Ddistribute_lib.py494 num_input_pipelines=1, argument
505 self._num_input_pipelines = num_input_pipelines
520 def num_input_pipelines(self): member in InputContext
546 self.input_pipeline_id, self.num_input_pipelines)
H A Dmirrored_strategy.py557 num_input_pipelines=num_workers,
589 num_input_pipelines=num_workers,
H A Dstrategy_common_test.py629 return d.shard(input_context.num_input_pipelines,
674 return dataset.shard(input_context.num_input_pipelines,
H A Dtpu_strategy.py955 num_input_pipelines=num_workers,
1021 num_input_pipelines=num_workers,
H A Dparameter_server_strategy_v2.py891 num_input_pipelines=num_input_pipelines_in_sync,
H A Ddistributed_table_test.py100 dataset = dataset.shard(input_context.num_input_pipelines,
H A Dinput_lib.py1132 num_workers = input_context.num_input_pipelines if input_context else len(
1145 input_context.num_input_pipelines,
H A Dcollective_all_reduce_strategy.py644 num_input_pipelines=self._num_workers,
H A Dstrategy_test_lib.py345 input_context.num_input_pipelines)
H A Dtpu_strategy_test.py292 dataset = dataset.shard(input_context.num_input_pipelines,
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.distribute.-input-context.pbtxt10 name: "num_input_pipelines"
19 …argspec: "args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.distribute.-input-context.pbtxt10 name: "num_input_pipelines"
19 …argspec: "args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'…