This is the complete list of members for test_jit.TestBatched, including all inherited members.
__init__(self, method_name='runTest') (defined in common_utils.TestCase) | common_utils.TestCase | |
assertAlmostEqual(self, x, y, places=None, msg=None, delta=None, allow_inf=None) (defined in common_utils.TestCase) | common_utils.TestCase | |
assertEqual(self, x, y, prec=None, message='', allow_inf=False) (defined in common_utils.TestCase) | common_utils.TestCase | |
assertExpected(self, s, subname=None) (defined in common_utils.TestCase) | common_utils.TestCase | |
assertExpectedInline(self, actual, expect, skip=0) (defined in expecttest.TestCase) | expecttest.TestCase | |
assertExpectedRaises(self, exc_type, callable, args, kwargs) (defined in common_utils.TestCase) | common_utils.TestCase | |
assertLeaksNoCudaTensors(self, name=None) (defined in common_utils.TestCase) | common_utils.TestCase | |
assertNotEqual(self, x, y, prec=None, message='') (defined in common_utils.TestCase) | common_utils.TestCase | |
assertObjectIn(self, obj, iterable) (defined in common_utils.TestCase) | common_utils.TestCase | |
assertRaisesRegex (defined in common_utils.TestCase) | common_utils.TestCase | static |
assertRegex (defined in common_utils.TestCase) | common_utils.TestCase | static |
assertTensorsSlowEqual(self, x, y, prec=None, message='') (defined in common_utils.TestCase) | common_utils.TestCase | |
assertWarns(self, callable, msg='') (defined in common_utils.TestCase) | common_utils.TestCase | |
assertWarnsRegex(self, callable, regex, msg='') (defined in common_utils.TestCase) | common_utils.TestCase | |
genSparseTensor(self, size, sparse_dim, nnz, is_uncoalesced, device='cpu') (defined in common_utils.TestCase) | common_utils.TestCase | |
longMessage (defined in expecttest.TestCase) | expecttest.TestCase | static |
maxDiff (defined in common_utils.TestCase) | common_utils.TestCase | static |
precision (defined in common_utils.TestCase) | common_utils.TestCase | static |
rand_batch(self, dims) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
safeCoalesce(self, t) (defined in common_utils.TestCase) | common_utils.TestCase | |
safeToDense(self, t) (defined in common_utils.TestCase) | common_utils.TestCase | |
setUp(self) (defined in common_utils.TestCase) | common_utils.TestCase | |
test_batch_argmax(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_cat(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_elementwise_binary(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_elementwise_unary(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_index_select(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_matmul(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_mm(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_select(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_softmax(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_sum(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_topk(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_view(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_batch_where(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_beam_search(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_create_batchtensor(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_for(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_greedy_search(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_if_else(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_if_else_with_scalar(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_if_noelse(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_if_noelse_with_scalar(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_lstm(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
test_while(self) (defined in test_jit.TestBatched) | test_jit.TestBatched | |
wrap_with_cuda_memory_check(self, method) (defined in common_utils.TestCase) | common_utils.TestCase |