Searched refs:compute_device (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 957 compute_device=True): argument
993 compute_device=compute_device)
1003 compute_device=compute_device)
1014 compute_device=True): argument
1021 compute_device)
1044 compute_device=compute_device)
1060 compute_device=True): argument
1080 name, attrs, op_def, compute_device)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | func_graph.py | 509 compute_device=True): argument
528 compute_device)
542 compute_device=True): argument
576 attrs, op_def, compute_device)
600 compute_device)
|
| D | ops.py | 3441 compute_device=True): argument
3484 attrs, op_def, compute_device)
3495 compute_device=True): argument
3554 self._create_op_helper(ret, compute_device=compute_device)
3557 def _create_op_from_tf_operation(self, c_op, compute_device=True): argument
3586 self._create_op_helper(ret, compute_device=compute_device)
3589 def _create_op_helper(self, op, compute_device=True): argument
3628 if compute_device:
3696 self._create_op_from_tf_operation(c_op, compute_device=compute_devices)
|
| D | function.py | 814 compute_device=True): argument
826 compute_device=compute_device)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 272 compute_device='CPU', argument
281 if 'CPU' in compute_device:
591 strategy, compute_device='CPU', variable_device='CPU', num_gpus=0)
597 strategy, compute_device='GPU', variable_device='GPU', num_gpus=1)
603 strategy, compute_device='GPU', variable_device='CPU', num_gpus=2)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-graph.pbtxt | 71 …nput_types\', \'name\', \'attrs\', \'op_def\', \'compute_shapes\', \'compute_device\'], varargs=No…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-graph.pbtxt | 71 …nput_types\', \'name\', \'attrs\', \'op_def\', \'compute_shapes\', \'compute_device\'], varargs=No…
|
| D | tensorflow.__internal__.-func-graph.pbtxt | 152 …nput_types\', \'name\', \'attrs\', \'op_def\', \'compute_shapes\', \'compute_device\'], varargs=No…
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1023 op_type, inputs, op_dtypes, attrs=attrs, compute_device=True)
|