Searched defs:train_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | builder_impl.py | 236 train_op=None, argument
309 train_op=None, argument
439 def _add_collections(self, assets_collection, main_op, train_op): argument
494 def _add_train_op(self, train_op): argument
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | file_input_yielder.cc | 106 for (const auto& train_op : train_ops) { in NextItem() local
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer_test.py | 89 def _run(self, train_op, sess): argument
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder_test.cc | 181 CollectionDef train_op; in TEST_F() local
201 CollectionDef train_op; in TEST_F() local
258 CollectionDef train_op; in TEST_F() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_utils_test.py | 203 def __new__(cls, mode, loss, train_op, scaffold=None): argument
|
| /external/tensorflow/tensorflow/python/profiler/ |
| D | model_analyzer_test.py | 587 def _trainLoop(self, train_op, train_steps, time_dir, time_step, argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 612 train_op = lambda: control_flow_util.smart_cond( function
|