Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dshared_variable_creator.py34 def make_fn(shared_variable_store, device_id): argument
67 if canonical_name not in shared_variable_store:
68 shared_variable_store[canonical_name] = []
69 shared_variable_store[canonical_name].append(v)
80 v = shared_variable_store[canonical_name][variable_index]
H A Dshared_variable_creator_test.py48 shared_variable_store = {}
52 creator_fn = shared_variable_creator.make_fn(shared_variable_store, i)
H A Dmirrored_run.py175 shared_variable_store = {}
184 shared_variable_store, index)