Searched refs:bodyb (Results 1 – 1 of 1) sorted by relevance
325 std::unique_ptr<XlaBuilder> bodyb = in InvertDiagonalBlocks() local329 Parameter(bodyb.get(), 0, tuple_shape, "InvertDiagBodyTuple"); in InvertDiagonalBlocks()335 auto zero = ConstantR0<int32>(bodyb.get(), 0); in InvertDiagonalBlocks()355 Tuple(bodyb.get(), {next_i, body_out, body_input}); in InvertDiagonalBlocks()357 TF_ASSIGN_OR_RETURN(auto body, bodyb->Build()); in InvertDiagonalBlocks()