Searched refs:IsBroadcastingRequired (Results 1 – 10 of 10) 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()
67 bool IsBroadcastingRequired() const { return broadcasting_required_; } in IsBroadcastingRequired() function
89 bool IsBroadcastingRequired() const { return broadcasting_required_; } in IsBroadcastingRequired() function
94 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run()126 const bool should_bcast = bcast.IsBroadcastingRequired();181 const bool should_bcast = bcast.IsBroadcastingRequired();363 (!bcast.IsBroadcastingRequired() || is_full_broadcast) &&375 } else if (!bcast.IsBroadcastingRequired()) {536 (!bcast.IsBroadcastingRequired() || is_full_broadcast) &&548 } else if (!bcast.IsBroadcastingRequired()) {
192 const bool should_bcast = bcast.IsBroadcastingRequired(); in operator ()()
359 const bool should_bcast = bcast.IsBroadcastingRequired(); in operator ()()
87 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run()266 if (!bcast.IsBroadcastingRequired() || out->shape() == in_y.shape()) {357 if (!bcast.IsBroadcastingRequired()) {
65 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run()
215 const int a_batch_idx = bcast.IsBroadcastingRequired() in Run()218 const int b_batch_idx = bcast.IsBroadcastingRequired() in Run()
164 if (bcast.IsBroadcastingRequired()) { in createMatMulOps()