| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy_test.py | 338 def model_fn(): function
360 result = distribution.extended.call_for_each_replica(model_fn)
376 def model_fn(): function
382 in_model_fn = distribution.extended.call_for_each_replica(model_fn)
390 in_model_fn = distribution.extended.call_for_each_replica(model_fn)
398 def model_fn(): function
403 result = distribution.extended.call_for_each_replica(model_fn)
412 def model_fn(): function
418 return distribution.extended.call_for_each_replica(model_fn)
434 def model_fn(): function
[all …]
|
| D | mirrored_variable_test.py | 104 def model_fn(): function
111 v2 = distribution.extended.call_for_each_replica(model_fn)
122 def model_fn(): function
132 distribution.extended.call_for_each_replica(model_fn))
138 def model_fn(): function
147 result = distribution.extended.call_for_each_replica(model_fn)
152 def model_fn(): function
158 result = distribution.extended.call_for_each_replica(model_fn)
163 def model_fn(): function
171 result = distribution.extended.call_for_each_replica(model_fn)
[all …]
|
| D | single_loss_example.py | 65 def model_fn(x): function
79 return model_fn, dataset_fn, layer
102 def model_fn(x): function
121 return model_fn, dataset_fn, batchnorm
|
| /external/tensorflow/tensorflow/python/estimator/ |
| D | BUILD | 28 ":model_fn",
60 name = "model_fn",
61 srcs = ["model_fn.py"],
104 ":model_fn",
120 ":model_fn",
133 ":model_fn",
146 ":model_fn",
163 ":model_fn",
183 ":model_fn",
211 ":model_fn",
[all …]
|
| D | model_fn.py | 26 from tensorflow_estimator.python.estimator import model_fn
30 model_fn.__all__ = [s for s in dir(model_fn) if not s.startswith('__')]
32 from tensorflow_estimator.python.estimator.model_fn import *
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | checkpoint_input_pipeline_hook_test.py | 36 from tensorflow_estimator.python.estimator import model_fn
58 return model_fn.EstimatorSpec(
84 est = estimator.Estimator(model_fn=self._model_fn)
99 est = estimator.Estimator(model_fn=self._model_fn)
112 est = estimator.Estimator(model_fn=self._model_fn)
128 est = estimator.Estimator(model_fn=self._model_fn)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | test_example.py | 42 def model_fn(x): function
56 return model_fn, dataset_fn, layer
79 def model_fn(x): function
98 return model_fn, dataset_fn, batchnorm
|
| D | minimize_loss_test.py | 95 model_fn, dataset_fn, layer = minimize_loss_example(
102 model_fn, args=(inputs,)))
138 model_fn, dataset_fn, layer = minimize_loss_example(
146 model_fn, args=(iterator.get_next(),)))
190 model_fn, dataset_fn, _ = minimize_loss_example(
197 model_fn, args=(inputs,)))
251 model_fn, dataset_fn, batchnorm = batchnorm_example(
262 model_fn, args=(inputs,)))
339 def model_fn(inputs): function
372 model_fn, args=(inputs,)))
[all …]
|
| D | keras_correctness_test_base.py | 256 model_fn, argument
261 model = model_fn(
489 model_fn=self.get_model,
495 model_fn=self.get_model,
583 model_fn=self.get_model,
588 model_fn=self.get_model,
|
| D | mirrored_variable_test.py | 75 def model_fn(features): function
98 model_fn, args=(features,))
|
| D | keras_optimizer_v2_test.py | 62 def model_fn(): function
74 model_fn)
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_architectures_test.py | 76 def test_basic_saving_and_loading(self, model_fn): argument
79 if 'subclassed_in_functional' in model_fn.__name__:
85 elif ('subclassed' in model_fn.__name__ and save_format == 'h5'):
90 model_data = model_fn()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint_test.py | 44 from tensorflow_estimator.python.estimator import model_fn as model_fn_lib
65 def model_fn(features, labels, mode, params): function
114 model_fn=model_fn,
164 model_fn=model_fn,
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | ram_file_system_test.py | 26 from tensorflow.python.estimator.model_fn import EstimatorSpec
101 def model_fn(features, labels, mode, params): function
124 estimator = Estimator(config=config, model_fn=model_fn)
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | vectorized_map_test.py | 29 def model_fn(arg): function
40 per_example_gradients = tf.vectorized_map(model_fn, (inputs, labels))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.estimator.-estimator-spec.pbtxt | 3 is_instance: "<class \'tensorflow_estimator.python.estimator.model_fn.EstimatorSpec\'>"
4 is_instance: "<class \'tensorflow_estimator.python.estimator.model_fn.EstimatorSpec\'>"
|
| D | tensorflow.estimator.-estimator.pbtxt | 15 name: "model_fn"
24 …argspec: "args=[\'self\', \'model_fn\', \'model_dir\', \'config\', \'params\', \'warm_start_from\'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.estimator.-estimator-spec.pbtxt | 3 is_instance: "<class \'tensorflow_estimator.python.estimator.model_fn.EstimatorSpec\'>"
4 is_instance: "<class \'tensorflow_estimator.python.estimator.model_fn.EstimatorSpec\'>"
|
| D | tensorflow.estimator.tpu.-t-p-u-estimator-spec.pbtxt | 4 is_instance: "<class \'tensorflow_estimator.python.estimator.model_fn._TPUEstimatorSpec\'>"
5 is_instance: "<class \'tensorflow_estimator.python.estimator.model_fn.TPUEstimatorSpec\'>"
|
| D | tensorflow.estimator.-estimator.pbtxt | 14 name: "model_fn"
23 …argspec: "args=[\'self\', \'model_fn\', \'model_dir\', \'config\', \'params\', \'warm_start_from\'…
|
| D | tensorflow.estimator.tpu.-t-p-u-estimator.pbtxt | 15 name: "model_fn"
24 …argspec: "args=[\'self\', \'model_fn\', \'model_dir\', \'config\', \'params\', \'use_tpu\', \'trai…
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | optimizer_benchmarks_test.py | 30 def model_fn(): function
44 return model_fn, x_train, y_train
|
| D | benchmark_util.py | 98 def measure_performance(model_fn, argument
172 model = model_fn()
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 125 model_fn = def_function.function(model)
126 func_output = model_fn(positive)
128 func_output = model_fn(negative)
155 model_fn = def_function.function(model)
156 func_output = model_fn(positive)
158 func_output = model_fn(negative)
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | applications_test.py | 138 def _get_output_shape(model_fn): argument
139 model = model_fn()
|