Home
last modified time | relevance | path

Searched refs:shared_variable_store (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dshared_variable_creator.py38 def make_fn(shared_variable_store, device_id): argument
71 if canonical_name not in shared_variable_store:
72 shared_variable_store[canonical_name] = []
73 shared_variable_store[canonical_name].append(v)
84 v = shared_variable_store[canonical_name][variable_index]
Dshared_variable_creator_test.py52 shared_variable_store = {}
56 creator_fn = shared_variable_creator.make_fn(shared_variable_store, i)
Dmirrored_run.py150 shared_variable_store = {}
157 shared_variable_store, index)