Searched refs:CreateDiagonal (Results 1 – 1 of 1) sorted by relevance
31 xla::XlaOp CreateDiagonal(xla::XlaOp input, int64 last_dim_size, in CreateDiagonal() function107 CreateDiagonal(input, size, /*other_dims=*/{}, ctx->input_xla_type(0)); in Compile()184 xla::XlaOp diag = CreateDiagonal(input, last_dim_size, other_dims, in Compile()