Searched refs:last_blocks (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 96 auto last_blocks = in DiagonalBlocks() local
101 last_blocks = in DiagonalBlocks()
102 Pad(last_blocks, Zero(builder, shape.element_type()), config); in DiagonalBlocks()
110 last_blocks = ConcatInDim(builder, {last_blocks, eye}, ndims - 1); in DiagonalBlocks()
114 TF_ASSIGN_OR_RETURN(Shape blocks_shape, builder->GetShape(last_blocks)); in DiagonalBlocks()
119 last_blocks = Reshape(last_blocks, last_blocks_dims); in DiagonalBlocks()
124 ConcatInDim(builder, {diag_blocks, last_blocks}, ndims - 2); in DiagonalBlocks()
126 diag_blocks = last_blocks; in DiagonalBlocks()
|