Searched defs:initial_values (Results 1 – 5 of 5) sorted by relevance
28 absl::Span<const XlaOp> initial_values, absl::string_view name, in WhileLoopHelper()85 absl::Span<const XlaOp> initial_values, absl::string_view name, in ForEachIndex()
604 absl::Span<const XlaOp> initial_values, // in WhileLoopFn()
249 const xla::XlaOp& initial_values, in BuildWhile()
301 StatusOr<std::vector<XlaOp>> Sweeps(absl::Span<const XlaOp> initial_values, in Sweeps()
709 fn initial_values() { in initial_values() function