Searched full:_shape_as_tensor (Results 1 – 17 of 17) sorted by relevance
| /aosp_15_r20/external/executorch/exir/ |
| H A D | control_flow.py | 85 # TODO _shape_as_tensor should work with functional tensor but currently not.
91 proxy = torch.ops.aten._shape_as_tensor.default(x.proxy)
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | operators.py | 31 return torch._shape_as_tensor(x)
|
| H A D | symbolic_opset9.py | 320 @_onnx_symbolic("aten::_shape_as_tensor")
321 def _shape_as_tensor(g: jit_utils.GraphContext, input): function
5393 shape_tensor = _shape_as_tensor(g, self)
5425 cum_adv_index_shape_tensor = _shape_as_tensor(g, cum_adv_index)
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _shape_functions.py | 1086 def _shape_as_tensor(self: List[int]) -> List[int]: function
1420 "aten::_shape_as_tensor(Tensor self) -> Tensor", _shape_as_tensor
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 44 - aten::_shape_as_tensor
731 aten::_shape_as_tensor: 10
|
| H A D | tensor_ops.py | 96 a._shape_as_tensor(),
|
| H A D | model_ops.yaml | 44 aten::_shape_as_tensor: 16
|
| /aosp_15_r20/external/executorch/test/end2end/ |
| H A D | test_end2end.py | 653 …# Fail with error: Missing out variants: {'aten::select', 'aten::_shape_as_tensor', 'aten::tensor_…
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | serialized_shape_function_registry.cpp | 2901 def _shape_as_tensor(self: List[int]) -> List[int]:
3320 {"aten::_shape_as_tensor(Tensor self) -> Tensor", "_shape_as_tensor"}, in GetShapeFunctionMappings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | constant_fold.cpp | 438 updated_val = at::_shape_as_tensor(inputTensorValues[0]); in runTorchBackendForOnnx()
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | torch.py | 99 torch._shape_as_tensor,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | allowlist_for_publicAPI.json | 1287 "_shape_as_tensor",
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_tensor.py | 1835 shape_tensor = _shape_as_tensor(shape, row_partition_tensors[0].dtype)
2979 def _shape_as_tensor(shape, dtype): function
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1555 "torch._shape_as_tensor",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.cpp | 366 Tensor _shape_as_tensor(const Tensor& self) { in _shape_as_tensor() function
|
| H A D | native_functions.yaml | 309 - func: _shape_as_tensor(Tensor self) -> Tensor
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_misc.py | 5151 return 1 + torch._shape_as_tensor(x)[0]
|