Searched refs:_XLA_SCOPE_KEY (Results 1 – 1 of 1) sorted by relevance
29 _XLA_SCOPE_KEY = ("__xla_scope",) variable137 xla_scope_counter = ops.get_collection(_XLA_SCOPE_KEY)140 ops.add_to_collection(_XLA_SCOPE_KEY, xla_scope_counter)