Searched refs:OMITTED_INDICES_KEY (Results 1 – 2 of 2) sorted by relevance
35 OMITTED_INDICES_KEY = "omitted" variable260 annotations[offset + i] = {OMITTED_INDICES_KEY: copy.copy(curr_indices)}357 elif OMITTED_INDICES_KEY in annot[r]:358 indices_key = OMITTED_INDICES_KEY374 OMITTED_INDICES_KEY in annot[prev_r]396 OMITTED_INDICES_KEY in annot[prev_r]
328 self.assertEqual({tensor_format.OMITTED_INDICES_KEY: [0, 2, 0]},338 self.assertEqual({tensor_format.OMITTED_INDICES_KEY: [10, 2, 0]},