Searched refs:all_indices (Results 1 – 6 of 6) sorted by relevance
89 all_indices = [93 np.random.shuffle(all_indices)94 indices = np.array(all_indices[:num_updates])
448 std::set<int64> all_indices; in RestoreInternal() local458 if (temp >= 0 && all_indices.find(temp) != all_indices.end()) { in RestoreInternal()464 all_indices.insert(temp); in RestoreInternal()479 if (all_indices.find(temp) != all_indices.end()) { in RestoreInternal()485 all_indices.insert(temp); in RestoreInternal()
71 all_indices = array_ops.reshape(indices + increments, [-1])75 array_ops.gather(value, all_indices),
286 all_indices=indices,434 init_values = [True, pfor_input.pfor.all_indices] + inputs + output_tas1027 all_indices=None, argument1057 assert all_indices is not None1058 self.all_indices = (1059 math_ops.range(loop_len) if all_indices is None else all_indices)1062 self._conversion_map[loop_var] = wrap(self.all_indices, True)1774 if indices == pfor_input.pfor.all_indices and axis == 0:2446 all_indices = pfor_input.pfor.all_indices2462 value = array_ops.gather(value, all_indices)
118 all_indices = [123 np.random.shuffle(all_indices)124 indices = np.array(all_indices[:num_updates])
365 all_indices = sorted(flatten([t_i["indices"] for t_i in taken]))373 self.assertAllEqual(all_indices, expected_indices)