Searched defs:TensorWrapper (Results 1 – 25 of 60) sorted by relevance
123
| /external/pytorch/aten/src/ATen/functorch/ |
| D | TensorWrapper.h | 35 struct TORCH_API TensorWrapper : public c10::TensorImpl { struct
46 const Tensor& value() const { in value()
49 std::optional<int64_t> level() const { in level()
75 // TensorWrapper receives a boolean flag on whether or not the Grad Interpreter argument
95 // In practice this isn't a problem: when we're constructing TensorWrapper in argument
|
| /external/executorch/backends/qualcomm/builders/ |
| D | op_skip_ops.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
43 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_conv2d.py | 108 nodes_to_wrappers: Dict[str, PyQnnWrapper.TensorWrapper],
268 nodes_to_wrappers: Dict[str, PyQnnWrapper.TensorWrapper],
|
| D | node_visitor.py | 260 nodes_to_wrappers: Dict[str, Dict[int, PyQnnWrapper.TensorWrapper]],
288 nodes_to_wrappers: Dict[str, Dict[int, PyQnnWrapper.TensorWrapper]],
357 nodes_to_wrappers: Dict[str, Dict[int, PyQnnWrapper.TensorWrapper]],
|
| D | op_relu.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_linear.py | 34 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_rsqrt.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_index_put.py | 21 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_sigmoid.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_sqrt.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_reshape.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_bmm.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_sub.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_hardswish.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_ceil.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_mul.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_upsample_bilinear2d.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_matmul.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_squeeze.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_div.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_add.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_gelu.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_tanh.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_upsample_nearest2d.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
| D | op_unsqueeze.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper],
|
123