Home
last modified time | relevance | path

Searched refs:batch_dim_ids (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dqr.cc81 std::vector<int64> batch_dim_ids(batch_dims.size()); in House() local
82 std::iota(batch_dim_ids.begin(), batch_dim_ids.end(), 0); in House()
115 *v = e_k + Div(x_after_k, divisor, /*broadcast_dimensions=*/batch_dim_ids); in House()