Searched refs:use_default_device (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | scan_ops.py | 36 use_default_device=None): argument
127 if use_default_device is not None:
134 use_default_device=use_default_device,
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | scan_test.py | 267 tf_api_version=2, mode="eager", use_default_device=[True, False]))
268 def testUseDefaultDevice(self, use_default_device): argument
285 dataset, np.int64(1), scan_fn, use_default_device=use_default_device)
288 if use_default_device:
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.h | 75 bool use_default_device = true; member
104 bool use_default_device() const { return use_default_device_; } in use_default_device() function
116 use_default_device_(params.use_default_device), in FunctionMetadata()
|
| D | iterator_ops.cc | 685 params.use_default_device = false; in ReduceDatasetOp()
|
| D | captured_function.cc | 596 inst_opts.default_device_to_target = metadata_->use_default_device(); in Instantiate()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | scan_dataset_op.cc | 40 params.use_default_device = use_default_device_; in ScanDatasetOp()
79 bool preserve_cardinality, bool use_default_device) in Dataset() argument
88 use_default_device_(use_default_device) { in Dataset()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ScanDataset.pbtxt | 54 name: "use_default_device"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ScanDataset.pbtxt | 107 name: "use_default_device"
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_deprecated_py2.py | 664 …return scan_ops._ScanDataset(ds, init_state, body, use_default_device=False) # pylint:disable=pro…
|
| D | control_flow.py | 710 …return scan_ops._ScanDataset(ds, init_state, body, use_default_device=False) # pylint:disable=pro…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3913 …, \'output_types\', \'output_shapes\', \'preserve_cardinality\', \'use_default_device\', \'name\']…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3913 …, \'output_types\', \'output_shapes\', \'preserve_cardinality\', \'use_default_device\', \'name\']…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 44033 name: "use_default_device"
|