Searched refs:cache_idx (Results 1 – 2 of 2) sorted by relevance
768 def _save_tensor_value_to_cache_op(self, graph, cache_idx, updates): argument772 indices = constant_op.constant([cache_idx])849 for cache_idx in range(0, len(self._cache_idx_to_tensor_idx)):850 tensor_idx = self._cache_idx_to_tensor_idx[cache_idx]851 line = '%d %d\n'%(cache_idx, tensor_idx)1163 cache_idx = len(self._traced_tensorname_to_cache_idx_map)1164 self._traced_tensorname_to_cache_idx_map[tensor_name] = cache_idx1497 cache_idx = self._traced_tensorname_to_cache_idx_map[tensor_name]1499 cache_idx,
438 int cache_idx = 0, trans_idx = n_cached_colors; in merge_colors() local440 if (cache_idx < n_cached_colors && in merge_colors()442 cached_colors[cache_idx] <= colors[trans_idx])) { in merge_colors()443 colors[i] = cached_colors[cache_idx++]; in merge_colors()