Home
last modified time | relevance | path

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

/external/libcxx/test/std/experimental/simd/simd.cons/
Dbroadcast.pass.cpp75 void test_broadcast() { in test_broadcast() function
83 test_broadcast<ex::native_simd<int>>(); in main()
84 test_broadcast<ex::fixed_size_simd<int, 4>>(); in main()
85 test_broadcast<ex::fixed_size_simd<int, 15>>(); in main()
/external/pytorch/test/distributed/
Dtest_c10d_spawn_ucc.py84 def test_broadcast(self): member in TestDistributedNNFunctionsUcc
Dtest_control_collectives.py35 def test_broadcast(self) -> None: member in TestCollectives
Dtest_c10d_spawn_nccl.py105 def test_broadcast(self): member in TestDistributedNNFunctionsNccl
Dtest_nccl.py62 def test_broadcast(self, device, dtype): member in TestNCCL
Dtest_fake_pg.py100 def test_broadcast(self): member in TestFakePG
Dtest_c10d_spawn_gloo.py229 def test_broadcast(self): member in TestDistributedNNFunctionsGloo
Dtest_multi_threaded_pg.py159 def test_broadcast(self): member in TestCollectivesWithBaseClass
Dtest_c10d_functional_native.py383 def test_broadcast(self) -> None: member in TestWithNCCL
Dtest_functional_api.py235 def test_broadcast(self, device): member in TestTraceableCollectives
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_random_test.py105 def test_broadcast(self, low_shape, high_shape, size): member in UniformTest
/external/pytorch/test/inductor/
Dtest_torchinductor_strided_blocks.py195 def test_broadcast( member in TritonBlockPointerTest
/external/pytorch/test/torch_np/numpy_tests/lib/
Dtest_shape_base_.py130 def test_broadcast(self): member in TestTakeAlongAxis
157 def test_broadcast(self): member in TestPutAlongAxis
/external/pytorch/test/
Dtest_tensorexpr.py210 def test_broadcast(self): member in TestTensorExprFuser
Dtest_jit_fuser_te.py331 def test_broadcast(self): member in TestTEFuser
Dtest_torch.py1053 def test_broadcast(self, fn, device): member in TestTorchDeviceType
/external/pytorch/torch/testing/_internal/distributed/
Ddistributed_test.py2078 def test_broadcast(self): member in DistributedTest._DistTestBase