Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-input-context.pbtxt6 name: "input_pipeline_id"
19 …argspec: "args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-input-context.pbtxt6 name: "input_pipeline_id"
19 …argspec: "args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'…
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy.py360 input_pipeline_id = multi_worker_util.id_in_cluster(
365 input_pipeline_id = 0
369 input_pipeline_id=input_pipeline_id,
381 input_pipeline_id = multi_worker_util.id_in_cluster(
386 input_pipeline_id = 0
391 input_pipeline_id=input_pipeline_id,
Dinput_lib_type_spec_test.py287 input_context.input_pipeline_id)
426 dataset = dataset.shard(ctx.num_input_pipelines, ctx.input_pipeline_id)
538 input_context.input_pipeline_id)
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)
Dinput_lib_test.py88 input_pipeline_id=i,
1029 if ctx.input_pipeline_id == 0:
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)
1590 np.full(4, input_context.input_pipeline_id))
1633 np.full(4, input_context.input_pipeline_id))
1664 (2, 5)) * (input_context.input_pipeline_id + 1))
1702 (2, 5)) * (input_context.input_pipeline_id + 1))
1744 np.arange(1, 10) * (input_context.input_pipeline_id + 1))
[all …]
Ddistribute_lib_test.py580 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
582 self.assertEqual(1, input_context.input_pipeline_id)
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)
Ddistribute_lib.py495 input_pipeline_id=0, argument
506 self._input_pipeline_id = input_pipeline_id
515 def input_pipeline_id(self): member in InputContext
546 self.input_pipeline_id, self.num_input_pipelines)
Dmirrored_strategy.py558 input_pipeline_id=i,
590 input_pipeline_id=i,
Dstrategy_common_test.py630 input_context.input_pipeline_id)
675 input_context.input_pipeline_id)
Dstrategy_test_lib.py348 input_context.input_pipeline_id)
350 self.assertEqual(worker_id_counter[0], input_context.input_pipeline_id)
Dparameter_server_strategy_v2.py892 input_pipeline_id=input_pipeline_id_in_sync,
Ddistributed_table_test.py101 input_context.input_pipeline_id)
Dtpu_strategy.py956 input_pipeline_id=i,
1022 input_pipeline_id=i,
Dinput_lib.py1143 dataset = rebatch_fn(dataset, input_context.input_pipeline_id)
1146 input_context.input_pipeline_id,
Dcollective_all_reduce_strategy.py645 input_pipeline_id=self._id_in_cluster,
Dtpu_strategy_test.py293 input_context.input_pipeline_id)