Searched refs:_XLA_SCOPE_KEY (Results 1 – 1 of 1) sorted by relevance
27 _XLA_SCOPE_KEY = ("__xla_scope",) variable95 xla_scope_counter = ops.get_collection(_XLA_SCOPE_KEY)98 ops.add_to_collection(_XLA_SCOPE_KEY, xla_scope_counter)