Searched defs:WhileCondFn (Results 1 – 1 of 1) sorted by relevance
340 struct WhileCondFn { struct341 const int64_t num_boxes;342 const int64_t output_size;344 explicit WhileCondFn(int64_t num_boxes, int64_t output_size) in WhileCondFn() function347 StatusOr<xla::XlaOp> operator()(absl::Span<const xla::XlaOp> values, in operator ()()