Home
last modified time | relevance | path

Searched refs:rhs_make_tensor_kwargs (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/testing/_internal/opinfo/
Dcore.py1669 exclude_zero = op.rhs_make_tensor_kwargs.get("exclude_zero", False)
1769 rhs = make_arg(shape, **op.rhs_make_tensor_kwargs)
1810 exclude_zero = op.rhs_make_tensor_kwargs.get("exclude_zero", False)
1976 shape_rhs, noncontiguous=noncontiguous, **op.rhs_make_tensor_kwargs
1994 rhs = make_arg(shape, **op.rhs_make_tensor_kwargs)
1996 rhs_scalar = make_arg((), **op.rhs_make_tensor_kwargs).item()
2006 rhs_scalar = make_arg((), **op.rhs_make_tensor_kwargs).item()
2037 rhs = make_arg(shape, **op.rhs_make_tensor_kwargs)
2059 rhs = make_arg((1026,), noncontiguous=True, **op.rhs_make_tensor_kwargs)
2066 rhs = make_arg((789, 357), **op.rhs_make_tensor_kwargs)
[all …]
/external/pytorch/test/
Dtest_binary_ufuncs.py279 (1026,), device=device, dtype=dtype, **op.rhs_make_tensor_kwargs
301 (789, 357), device=device, dtype=dtype, **op.rhs_make_tensor_kwargs
325 shape, dtype=dtype, device=device, **op.rhs_make_tensor_kwargs
355 shape, dtype=dtype, device=device, **op.rhs_make_tensor_kwargs
382 shape, dtype=dtype, device=device, **op.rhs_make_tensor_kwargs
406 shape, dtype=dtype, device=device, **op.rhs_make_tensor_kwargs
434 shape, dtype=dtype, device=device, **op.rhs_make_tensor_kwargs
462 shape, dtype=dtype, device=device, **op.rhs_make_tensor_kwargs
486 make_tensor, (5,), device=device, **op.rhs_make_tensor_kwargs
490 make_tensor, (), device="cpu", **op.rhs_make_tensor_kwargs
[all …]
/external/pytorch/torch/testing/_internal/opinfo/definitions/
Dspecial.py234 rhs_make_tensor_kwargs=dict(low=-0.99),
/external/pytorch/torch/testing/_internal/
Dcommon_methods_invocations.py766 rhs = make_arg((S, S), **op.rhs_make_tensor_kwargs)
773 rhs = make_arg((S, S), **op.rhs_make_tensor_kwargs)
1118 rhs = make_arg((S, S), **op.rhs_make_tensor_kwargs)
11970 rhs_make_tensor_kwargs=dict(exclude_zero=False),
11988 rhs_make_tensor_kwargs=dict(exclude_zero=False),
12512 rhs_make_tensor_kwargs=dict(low=0),
12526 rhs_make_tensor_kwargs=dict(low=0),
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),
[all …]