Searched defs:_test_op (Results 1 – 4 of 4) sorted by relevance
313 def _test_op(self, mesh, op_call, *args, **kwargs): member in DistTensorOpsTest
73 def _test_op( member in TestSerialization
342 def _test_op(self, mesh: DeviceMesh, op_call, *args, **kwargs) -> None: member in DTensorTestBase
95 typedef enum _test_op { enum