Lines Matching +defs:val +defs:helper
1682 def helper(val, shape, dtype): argument
1873 def helper(dtype): function
1989 def helper(input_shape, batch1_shape, batch2_shape): function
2036 def helper(bias_shape): function
2237 …def helper(shape, ks, padding=0, dilation=1, ceil_mode=False, return_indices=False, test_ties=Fals… function
2311 def helper(size, memory_format): function
2337 def helper(shape): function
2393 def helper(shape, channels_last=False): function
2410 def helper(shape, eps=1, momentum=0.1, wts=False, training=False, channels_last=False, function
2682 …def helper(input_shape, normalized_shape, eps=1e-05, elementwise_affine=True, dtype=torch.float32): function
2742 …def helper(shape, eps=1, momentum=0.1, wts=False, channels_last=False, track_running_stats=True, t… function
2919 def helper(dim, layer='linear', dtype=torch.float32): function
2990 def helper(input_shape, wt_shape, function
3063 def helper(input_shape, wt_shape, function
3131 def helper(shape): function
3156 def helper(shape): function
3180 def helper(threshold, value, num_elems, inplace=False, requires_grad=True): function
3207 def helper(shape): function
3240 def helper(shape, value, xtype=torch.float32, ytype=None, ztype=None): function
3277 def helper(shape, value): function
3303 def helper(shape, value): function
3485 def helper(shape): function
3605 def helper(x_tensor, y_func, z_func, r_func=None): function
3683 def helper(binary_op): function
3698 def helper(operator): function
3790 def helper(shape, repeats): function
3890 def helper(shape, dtype=torch.float32, num_repeats=torch.Tensor(), dim=None): function
3907 def helper(dtype): function
4007 def helper(data, to_dtype): function
4025 def helper(data, to_dtype): function
4141 def helper(dtype): function
4198 def helper(x, return_inverse, return_counts): function
4218 def helper(x, dim, return_inverse, return_counts): function
4310 def helper(src_shape, dst_shape, src_dtype, dst_dtype): function
4399 def helper(dtype, noncontiguous, dim): function
4444 def helper(dtype): function
4469 def helper(dtype): function
4486 def helper(dtype): function
4504 def helper(dtype): function
4521 def helper(shape, dtype): function
4719 def helper(n, c): function
4736 def helper(shape, reduction): function
4765 def helper(shape, reduction): function
4821 def helper(shape, reduction): function
5070 def helper(shape): function
5083 def helper(shape): function
5094 def helper(shape): function
5107 def helper(shape): function
5118 def helper(shape): function
5131 def helper(shape): function
5142 def helper(shape): function
5155 def helper(shape): function
5166 def helper(shape): function
5179 def helper(shape): function
5208 def helper(n, c, h, w, reduction_type, dtype=torch.float32): function
5283 def helper(n, c, h, w, dtype=torch.float32): function
5408 def helper(shape): function
5473 def helper(fn, dim): function
5490 def helper(shape): function
5560 def helper(n, c, h, w): function
5668 def helper(n, c, h, w, dtype=torch.float32): function
5737 def helper(shape, dtype=torch.float32): function
5771 def helper(n, c, h, w): function
5824 def helper(shape): function
5924 def helper(): function
5968 def helper(shape, dim, keepdim): function
5990 def helper(shape, dim, keepdim): function
6012 def helper(n, c, h, w): function
6065 def helper(n, c, h, w): function
6087 def helper(n, c, h, w): function
6108 def helper(n, c, h, w): function
6178 def helper(shape, rounding_mode): function
6210 def helper(shape): function
6247 def helper(n, c): function
6260 def helper(x): function
6267 def helper(n, c): function
6307 def helper(N, C, H, W, memory_format): function
6340 def helper(N, C, H, W): function
6373 def helper(shape, output_size, scales, mode, align_corners=False): function
6418 def helper(shape_x, shape_y, shape_z): function
6446 def helper(shape, dtype=torch.float32): function
6489 def helper(shape): function
6501 def helper(shape): function
6513 def helper(shape): function
6525 def helper(shape): function
6537 def helper(shape): function
6549 def helper(shape): function
6564 def helper(shape): function
6579 def helper(shape): function
6611 def helper(shape_x, shape_y, shape_z): function
6633 def helper(shape): function
6649 def helper(shape): function
6669 def helper(shape): function
6689 def helper(shape, alpha=1.0, memory_format=torch.contiguous_format): function
6727 def helper(shape, dim=0): function
6750 def helper(shape, beta, threshold, dtype): function
6778 def helper(shape): function
6799 def helper(src_dtype, dst_dtype): function
6810 def helper(src_dtype, dst_dtype): function
6837 def helper(input_shape, out_shape, channels_last): function
6882 def helper(input_shape, out_shape, return_indices, dtype, channels_last=False): function
6924 def helper(shape, dtype=torch.float, contiguous=True): function
6964 def helper(shape, dtype=torch.float, contiguous=True): function
7030 def helper(shape): function
7042 def helper(shape, min_val, max_val, inplace=False): function
7072 def helper(shape, inplace=False, requires_grad=True): function
7164 def helper(shape): function
7182 def helper(shape, dtype): function
7197 def helper(shape): function
7223 … def helper(shape, dim, index, source_shape, alpha, x_dtype=torch.float32, idx_dtype=torch.int32): function
7305 def helper(shape, dims): function
7329 def helper(shape, dim, index, idx_dtype=torch.int32): function
7351 def helper(value, dim, index, idx_dtype=torch.int32): function
7368 def helper(n, d, m, idx): function
7401 def helper(shape, dim, idx_shape, idx_dtype=torch.int64): function
7456 def helper(shape, dim, idx_shape, src_shape, idx_dtype=torch.int64, do_add=True): function
7523 def helper(idx_dtype=torch.int64, do_add=True): function
7563 def helper(shape, dim, idx_shape, src_shape, idx_dtype=torch.int64, reduce_str="sum"): function
7607 def helper(shape, diag=0): function
7633 def helper(n): function
7648 def helper(shape, diag=0): function
7674 def helper(n, m, dtype): function
7697 def helper(shape, diag=0): function
7719 def helper(start, end, steps, dtype=torch.float32): function
7754 def helper(shape, dim, channels_last=False): function
7813 def helper(shape, x_shape, y_shape, cond_dtype=torch.bool, x_dtype=torch.float): function
7861 def helper(shape, mean=0.0, std=1.0): function
8071 def helper(shape, low, high, dtype=torch.int32): function
8094 def helper(shape, lamda, dtype=torch.float32): function
8120 def helper(shape, alpha, op_name, inplace): function
8162 def helper(alpha): function
8198 def helper(shape, op): function
8235 def helper(op): function
8253 def helper(shape, op, inplace, dtype=torch.float32): function
8280 def helper(shape): function
8298 def helper(probs, compare_mean, compare_var, num_samples=5, replacement=True): function
8336 def helper(x): function
8354 def helper(x, other): function
8377 def helper(x, other): function
8401 def helper(x, other): function
8425 def helper(dtype): function
8444 def helper(dtype): function
8467 def helper(dtype): function
9090 def helper(shape, padding, op, value=0): function
10400 def helper(stride, padding): function
10429 def helper(stride): function
10456 def helper(shape, in_channels=1, out_channels=1, kernel_size=3, groups=1): function
10499 def helper(N, C, H, W, groups, input_mem_format, weight_mem_format, permute_data): function
10537 def helper(m_cpu, memory_format): function
10858 def helper(dtype): function
10986 def helper(dtype): function
10996 def helper(dtype): function
11014 def helper(dtype): function
11030 def helper(dtype): function
11045 def helper(x_cpu): function
11075 def helper(x_cpu): function
11121 def helper(dtype): function
11341 def helper(dtype, accumulate, deterministic, num_tests=128): function
11410 def helper(device, dtype): function
11420 def helper(device, dtype): function