Searched refs:nd_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_dynamo/backends/ |
| D | tvm.py | 132 def to_torch_tensor(nd_tensor): argument
134 if nd_tensor.dtype == "bool":
138 return torch.from_numpy(nd_tensor.numpy())
139 return torch.utils.dlpack.from_dlpack(nd_tensor.to_dlpack())
|
| /external/pytorch/torch/testing/_internal/opinfo/ |
| D | core.py | 2559 nd_tensor = partial(
2597 nd_tensor = partial(
2618 nd_tensor(),
2623 yield SampleInput(nd_tensor(), norm="ortho")
2624 yield SampleInput(nd_tensor(), s=(8,))
2626 yield from (SampleInput(nd_tensor(), dim=dim) for dim in [-1, -2, -3, (0, -1)])
2629 nd_tensor(),
2634 yield SampleInput(nd_tensor(), norm="ortho")
2635 yield SampleInput(nd_tensor(), s=(6, 8) if not is_fp16_or_chalf else (4, 8))
2636 yield SampleInput(nd_tensor(), dim=0)
[all …]
|