Searched refs:condb (Results 1 – 1 of 1) sorted by relevance
194 std::unique_ptr<XlaBuilder> condb = in InvertDiagonalBlocks() local198 Parameter(condb.get(), 0, tuple_shape, "InvertDiagCondTuple"), 0); in InvertDiagonalBlocks()199 Lt(i, ConstantR0<int32>(condb.get(), block_size)); in InvertDiagonalBlocks()201 TF_ASSIGN_OR_RETURN(auto cond, condb->Build()); in InvertDiagonalBlocks()