Searched refs:IsBroadcastingRequired (Results 1 – 9 of 9) sorted by relevance
41 EXPECT_TRUE(bcast.IsBroadcastingRequired()); in TEST()54 EXPECT_FALSE(bcast.IsBroadcastingRequired()); in TEST()67 EXPECT_FALSE(bcast.IsBroadcastingRequired()); in TEST()80 EXPECT_TRUE(bcast1.IsBroadcastingRequired()); in TEST()89 EXPECT_TRUE(bcast2.IsBroadcastingRequired()); in TEST()
37 bool IsBroadcastingRequired() const { return broadcasting_required_; } in IsBroadcastingRequired() function
89 bool IsBroadcastingRequired() const { return broadcasting_required_; } in IsBroadcastingRequired() function
96 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run()133 const bool should_bcast = bcast.IsBroadcastingRequired();174 const bool should_bcast = bcast.IsBroadcastingRequired();334 if (!bcast.IsBroadcastingRequired()) {470 if (!bcast.IsBroadcastingRequired()) {557 const bool should_bcast = bcast.IsBroadcastingRequired();
87 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run()266 if (!bcast.IsBroadcastingRequired() || out->shape() == in_y.shape()) {357 if (!bcast.IsBroadcastingRequired()) {
157 if (!bcast.IsBroadcastingRequired()) { in Compute()
189 const bool should_bcast = bcast.IsBroadcastingRequired(); in operator ()()
214 const int a_batch_idx = bcast.IsBroadcastingRequired() in Run()217 const int b_batch_idx = bcast.IsBroadcastingRequired() in Run()
169 if (bcast.IsBroadcastingRequired()) { in createMatMulOps()