Searched refs:BroadcastTest (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | broadcast_test.cc | 34 class BroadcastTest : public HloTestBase {}; class
36 XLA_TEST_F(BroadcastTest, BroadcastScalarToScalar) { in XLA_TEST_F() argument
53 XLA_TEST_F(BroadcastTest, BroadcastScalarTo2D) { in XLA_TEST_F() argument
70 XLA_TEST_F(BroadcastTest, BroadcastVectorTo2D) { in XLA_TEST_F() argument
97 XLA_TEST_F(BroadcastTest, Broadcast2DTo2D) { in XLA_TEST_F() argument
114 XLA_TEST_F(BroadcastTest, Broadcast2DTo2DTranspose) { in XLA_TEST_F() argument
133 XLA_TEST_F(BroadcastTest, Broadcast2DTo3D) { in XLA_TEST_F() argument
151 TEST_F(BroadcastTest, Broadcast_R1_2_To_R4_2x2x3x3) { in TEST_F() argument
173 TEST_F(BroadcastTest, Broadcast_R1_1025_To_R4_3x3x3x1025) { in TEST_F() argument
203 XLA_TEST_F(BroadcastTest, Broadcast_R1_64_To_R4_32x64x7x7) { in XLA_TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | pow_test.cc | 100 TEST(PowOpModel, BroadcastTest) { in TEST() argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nccl_ops_test.py | 159 class BroadcastTest(NcclTestCase): class
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 433 TEST_F(DynamicDimensionInferenceTest, BroadcastTest) { in TEST_F() argument
|