Searched defs:initial_values (Results 1 – 6 of 6) 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()
589 absl::Span<const XlaOp> initial_values, // in WhileLoopFn()
76 unsigned initial_values; member in __anon565931fb0111::acp_entry
249 const xla::XlaOp& initial_values, in BuildWhile()
301 StatusOr<std::vector<XlaOp>> Sweeps(absl::Span<const XlaOp> initial_values, in Sweeps()
671 fn initial_values() { in initial_values() function