Searched refs:input_pipeline_id (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-input-context.pbtxt | 6 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/ |
| D | tensorflow.distribute.-input-context.pbtxt | 6 name: "input_pipeline_id"
19 …argspec: "args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'…
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib_test.py | 153 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
155 self.assertEqual(1, input_context.input_pipeline_id)
160 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
|
| D | collective_all_reduce_strategy.py | 341 input_pipeline_id = 0
343 input_pipeline_id = multi_worker_util.id_in_cluster(
347 input_pipeline_id=input_pipeline_id,
|
| D | parameter_server_strategy.py | 265 input_pipeline_id = multi_worker_util.id_in_cluster(
270 input_pipeline_id = 0
274 input_pipeline_id=input_pipeline_id,
|
| D | distribute_lib.py | 240 input_pipeline_id=0, argument
251 self._input_pipeline_id = input_pipeline_id
260 def input_pipeline_id(self): member in InputContext
|
| D | tpu_strategy.py | 271 input_pipeline_id=i,
|
| D | mirrored_strategy.py | 582 input_pipeline_id=i,
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | input_lib_test.py | 49 input_pipeline_id=i,
261 if ctx.input_pipeline_id == 0:
|
| D | strategy_test_lib.py | 296 input_context.input_pipeline_id)
298 self.assertEqual(worker_id_counter[0], input_context.input_pipeline_id)
|