| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | iterator_ops.py | 32 def _convert_external_state_policy_to_enum(external_state_policy): argument
33 if external_state_policy == "warn":
35 if external_state_policy == "ignore":
37 if external_state_policy == "fail":
42 external_state_policy))
49 def make_saveable_from_iterator(iterator, external_state_policy="fail"): argument
98 policy_enum = _convert_external_state_policy_to_enum(external_state_policy)
102 external_state_policy=policy_enum)
151 def __init__(self, estimator, external_state_policy="fail"): argument
189 external_state_policy)
[all …]
|
| D | data_service_ops.py | 659 external_state_policy = dataset.options().experimental_external_state_policy
660 if external_state_policy is None:
661 external_state_policy = ExternalStatePolicy.WARN
677 external_state_policy=external_state_policy.value)
|
| D | distribute.py | 346 strip_device_assignment=True, external_state_policy=policy)
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | serialization_utils.cc | 61 params.external_state_policy = in AsGraphDefMinimal()
82 if (serialization_ctx.external_state_policy() == in AsGraphDef()
86 if (serialization_ctx.external_state_policy() == in AsGraphDef()
|
| D | dataset_ops.cc | 85 params.external_state_policy = external_state_policy_; in Compute()
|
| D | iterator_ops.cc | 1143 params.external_state_policy = external_state_policy_; in Compute()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | DatasetToGraphV2.pbtxt | 12 name: "external_state_policy"
|
| D | RegisterDataset.pbtxt | 20 name: "external_state_policy"
|
| D | SerializeIterator.pbtxt | 24 name: "external_state_policy"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | DatasetToGraphV2.pbtxt | 12 name: "external_state_policy"
|
| D | RegisterDataset.pbtxt | 20 name: "external_state_policy"
|
| D | SerializeIterator.pbtxt | 24 name: "external_state_policy"
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 824 external_state_policy=policy)
894 external_state_policy=distribute_options.ExternalStatePolicy.FAIL): argument
896 iterator_resource, external_state_policy=external_state_policy.value)
|
| D | dataset_ops.py | 235 external_state_policy=distribute_options.ExternalStatePolicy.WARN): argument
251 if external_state_policy:
252 policy = external_state_policy.value
255 external_state_policy=policy,
282 self._as_serialized_graph(external_state_policy=distribute_options
3060 pb.external_state_policy = (
3076 pb.external_state_policy))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.-checkpoint-input-pipeline-hook.pbtxt | 8 …argspec: "args=[\'self\', \'estimator\', \'external_state_policy\'], varargs=None, keywords=None, …
|
| D | tensorflow.data.experimental.pbtxt | 165 …argspec: "args=[\'iterator\', \'external_state_policy\'], varargs=None, keywords=None, defaults=[\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.-checkpoint-input-pipeline-hook.pbtxt | 8 …argspec: "args=[\'self\', \'estimator\', \'external_state_policy\'], varargs=None, keywords=None, …
|
| D | tensorflow.data.experimental.pbtxt | 189 …argspec: "args=[\'iterator\', \'external_state_policy\'], varargs=None, keywords=None, defaults=[\…
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | data_service_ops.cc | 55 params.external_state_policy = external_state_policy_; in Compute()
|
| D | snapshot_dataset_op.cc | 878 params.external_state_policy = in MakeDataset()
1029 params.external_state_policy = in MakeDataset()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | data_service_ops_test.py | 501 def run_stateful(self, external_state_policy): argument
507 options.experimental_external_state_policy = external_state_policy
517 combinations.combine(external_state_policy=[
521 def testStatefulNoError(self, external_state_policy): argument
522 self.run_stateful(external_state_policy)
|
| D | auto_shard_dataset_test.py | 323 external_state_policy=distribute_options.ExternalStatePolicy.WARN)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.h | 534 switch (params_.external_state_policy) { in HandleCheckExternalStateStatus()
551 ExternalStatePolicy external_state_policy = ExternalStatePolicy::kWarn; member
580 ExternalStatePolicy external_state_policy() const { in external_state_policy() function
581 return params_.external_state_policy; in external_state_policy()
|
| D | dataset_options.proto | 177 ExternalStatePolicy external_state_policy = 6; field
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | dataset_test.py | 64 dataset._as_serialized_graph(external_state_policy=distribute_options
|