Searched refs:broadcast_dim (Results 1 – 4 of 4) sorted by relevance
79 for (int64& broadcast_dim : broadcast_dims) { in BroadcastTo()80 broadcast_dim = broadcast_shape_size - broadcast_dim - 1; in BroadcastTo()
137 int64 broadcast_dim = hlo->dimensions(dimension); in HandleBroadcast() local138 parent_->SetDynamicSize(hlo, {}, broadcast_dim, dynamic_size); in HandleBroadcast()
893 for (int64 broadcast_dim : broadcast_dims) { in ComputeArrayForElementwiseBinaryOp() local894 simulated_index[broadcast_dim] = IndexComponent::NotBroadcasted; in ComputeArrayForElementwiseBinaryOp()
1926 for (int broadcast_dim : {0, 1}) { in XLA_TEST_F()1930 Max(u, v, /*broadcast_dimensions=*/{broadcast_dim}); in XLA_TEST_F()