Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-input-context.pbtxt10 name: "num_input_pipelines"
19 …argspec: "args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-input-context.pbtxt10 name: "num_input_pipelines"
19 …argspec: "args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'…
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy.py363 num_input_pipelines = multi_worker_util.worker_count(
367 num_input_pipelines = 1
369 num_input_pipelines=num_input_pipelines,
384 num_input_pipelines = multi_worker_util.worker_count(
388 num_input_pipelines = 1
391 num_input_pipelines=num_input_pipelines,
Ddistribute_lib_test.py563 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
566 self.assertEqual(2, input_context.num_input_pipelines)
570 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
577 num_input_pipelines=1, input_pipeline_id=0, num_replicas_in_sync=42)
582 num_input_pipelines=3, input_pipeline_id=1, num_replicas_in_sync=42)
Dinput_lib_type_spec_test.py290 dataset = dataset.shard(input_context.num_input_pipelines,
356 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
417 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
479 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
Dstrategy_common_test.py204 return d.shard(input_context.num_input_pipelines,
249 return dataset.shard(input_context.num_input_pipelines,
Ddistribute_lib.py470 num_input_pipelines=1, argument
481 self._num_input_pipelines = num_input_pipelines
496 def num_input_pipelines(self): member in InputContext
522 self.input_pipeline_id, self.num_input_pipelines)
Dmirrored_strategy.py552 num_input_pipelines=num_workers,
583 num_input_pipelines=num_workers,
Dinput_lib_test.py87 num_input_pipelines=input_workers.num_workers,
1056 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
1176 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
1225 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
Dtpu_strategy.py897 num_input_pipelines=num_workers,
964 num_input_pipelines=num_workers,
Dinput_lib.py1001 num_workers = input_context.num_input_pipelines if input_context else len(
1015 input_context.num_input_pipelines,
Dcollective_all_reduce_strategy.py588 num_input_pipelines=self._num_workers,
Dstrategy_test_lib.py328 input_context.num_input_pipelines)
/external/tensorflow/tensorflow/python/keras/utils/
Ddataset_creator_test.py68 dataset = dataset.shard(input_context.num_input_pipelines,
/external/tensorflow/tensorflow/python/keras/integration_test/
Dmulti_worker_tutorial_test.py100 dataset = dataset.shard(input_context.num_input_pipelines,