Searched defs:max_sequence_lengths (Results 1 – 3 of 3) sorted by relevance
33 ShapeHandle step_ids, parent_ids, max_sequence_lengths, end_token; in __anon559769010102() local
32 const int32* max_sequence_lengths, in GatherTreeOpKernel()
52 const Tensor& max_sequence_lengths = ctx->input(2); in Compute() local