Searched refs:scores_input (Results 1 – 1 of 1) sorted by relevance
435 const xla::XlaOp scores_input = context->Input("scores"); in Compile() local451 xla::Sort({xla::Broadcast(scores_input, {4}), boxes}, in Compile()459 {scores_input, iota_indices}, in Compile()