Searched defs:train_op (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | builder_impl.py | 237 train_op=None, argument
310 train_op=None, argument
441 def _add_collections(self, assets_collection, main_op, train_op): argument
497 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 | 86 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 | 201 def __new__(cls, mode, loss, train_op, scaffold=None): argument
|
| /external/tensorflow/tensorflow/python/profiler/ |
| D | model_analyzer_test.py | 578 def _trainLoop(self, train_op, train_steps, time_dir, time_step, memory_dir, argument
|