Searched defs:WhileCondFn (Results 1 – 1 of 1) sorted by relevance
333 struct WhileCondFn { struct334 const int64 num_boxes;335 const int64 output_size;337 explicit WhileCondFn(int64 num_boxes, int64 output_size) in WhileCondFn() argument340 xla::StatusOr<xla::XlaOp> operator()(absl::Span<const xla::XlaOp> values, in operator ()()