Searched refs:minor_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 89 const int64 minor_dim = batch_dims.size(); in House() local
99 /*broadcast_dimensions=*/{minor_dim}); in House()
110 {minor_dim}); in House()
126 CreateScalarAddComputation(type, builder), {minor_dim}); in House()
209 const int64 minor_dim = batch_dims.size(); in QrBlock() local
212 minor_dim + 1); in QrBlock()
241 /*broadcast_dimensions=*/ConcatVectors(batch_dim_indices, {minor_dim})); in QrBlock()
255 minor_dim); in QrBlock()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5713 const int64_t minor_dim = batch_dims.size(); in House() local
5732 loc, x, gtk, GetI64ElementsAttr({minor_dim}, builder)); in House()
5736 loc, x_after_k_sq, zero, GetI64ElementsAttr({minor_dim}, builder)); in House()
5903 const int64_t minor_dim = num_batch_dims; in QRBlock() local
5907 builder->getI64IntegerAttr(minor_dim + 1)); in QRBlock()
5938 builder->getI64IntegerAttr(minor_dim)); in QRBlock()
|