Searched defs:parent_ids (Results 1 – 5 of 5) sorted by relevance
33 ShapeHandle step_ids, parent_ids, max_sequence_lengths, end_token; in __anon559769010102() local
31 const T* parent_ids, in GatherTreeOpKernel()
51 const Tensor& parent_ids = ctx->input(1); in Compute() local
33 u8 parent_ids[UNIPHIER_CLK_MUX_MAX_PARENTS]; member
129 def gather_tree_from_array(t, parent_ids, sequence_length): argument490 def _maybe_sort_array_beams(self, t, parent_ids, sequence_length): argument