Searched refs:tau_broadcast_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 250 std::vector<int64> tau_broadcast_dims(batch_dims.size()); in QrBlock() local
251 std::iota(tau_broadcast_dims.begin(), tau_broadcast_dims.end(), 0); in QrBlock()
259 Add(taus_zeros, tau, /*broadcast_dimensions=*/tau_broadcast_dims), in QrBlock()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5931 llvm::SmallVector<int64_t, 4> tau_broadcast_dims(batch_dims.size()); in QRBlock() local
5932 std::iota(tau_broadcast_dims.begin(), tau_broadcast_dims.end(), 0); in QRBlock()
5952 GetI64ElementsAttr(tau_broadcast_dims, builder), *builder), in QRBlock()
|