Home
last modified time | relevance | path

Searched refs:exclude_zero (Results 1 – 9 of 9) sorted by relevance

/external/pytorch/torch/testing/_internal/opinfo/
Dcore.py1626 op, device, dtype, requires_grad, exclude_zero, **kwargs argument
1634 exclude_zero=exclude_zero,
1649 exclude_zero=exclude_zero,
1669 exclude_zero = op.rhs_make_tensor_kwargs.get("exclude_zero", False)
1677 exclude_zero,
1693 exclude_zero=exclude_zero,
1701 exclude_zero=exclude_zero,
1744 op, *, device, dtype, requires_grad=False, exclude_zero=False argument
1765 exclude_zero=exclude_zero,
1774 op, *, device, dtype, requires_grad=False, exclude_zero=False argument
[all …]
/external/pytorch/torch/testing/
D_creation.py52 exclude_zero: bool = False,
261 if exclude_zero:
/external/pytorch/torch/testing/_internal/opinfo/definitions/
Dspecial.py50 exclude_zero = requires_grad and op_info.op == torch.special.i0e
56 exclude_zero=exclude_zero,
61 if requires_grad and not exclude_zero:
Dsparse.py515 exclude_zero=True,
/external/pytorch/test/inductor/
Dtest_unbacked_symints.py50 x = make_tensor(32, 4, device=device, dtype=torch.float32, exclude_zero=True)
Dtest_torchinductor.py2455 make_tensor((), device=self.device, dtype=dtype, exclude_zero=True),
2462 make_tensor(10, device=self.device, dtype=dtype, exclude_zero=True),
2476 100, device=self.device, dtype=dtype, exclude_zero=True
/external/pytorch/test/
Dtest_reductions.py383 t = make_tensor((5, 3, 4, 2), dtype=dtype, device=device, low=-2, high=2, exclude_zero=True)
392 …est_ref(op, make_tensor((2 ** 20,), dtype=dtype, device=device, low=-1, high=1, exclude_zero=True))
398 … t = make_tensor((32, 2 ** 16), dtype=dtype, device=device, low=-1, high=1, exclude_zero=True)
406 …t_ref(op, make_tensor((275000000,), dtype=dtype, device=device, low=-1, high=1, exclude_zero=True))
412 t = make_tensor((4, 4), dtype=dtype, device=device, low=-2, high=2, exclude_zero=True)
422 t = make_tensor((5,), dtype=dtype, device=device, exclude_zero=True)
Dtest_testing.py1468 …t = torch.testing.make_tensor(10_000, dtype=dtype, device=device, exclude_zero=True, low=-1, high=…
/external/pytorch/torch/testing/_internal/
Dcommon_methods_invocations.py649 yield SampleInput(make_arg(shape, exclude_zero=True), args=args, name=name)
1327 args = tuple(make_arg(arg, exclude_zero=True) if isinstance(arg, tuple) else arg
1334 args = tuple(make_arg(arg, exclude_zero=True) if isinstance(arg, tuple) else arg
11970 rhs_make_tensor_kwargs=dict(exclude_zero=False),
11988 rhs_make_tensor_kwargs=dict(exclude_zero=False),
12974 rhs_make_tensor_kwargs=dict(exclude_zero=True),),
12986 rhs_make_tensor_kwargs=dict(exclude_zero=True),
13014 rhs_make_tensor_kwargs=dict(exclude_zero=True),
13039 rhs_make_tensor_kwargs=dict(exclude_zero=True)),
13467 rhs_make_tensor_kwargs=dict(exclude_zero=True),
[all …]