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()
604 absl::Span<const XlaOp> initial_values, // in WhileLoopFn()
76 unsigned initial_values; member in __anon72d847140111::acp_entry
249 const xla::XlaOp& initial_values, in BuildWhile()
301 StatusOr<std::vector<XlaOp>> Sweeps(absl::Span<const XlaOp> initial_values, in Sweeps()
693 fn initial_values() { in initial_values() function