Home
last modified time | relevance | path

Searched refs:_shape_as_tensor (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/torch/onnx/
Doperators.py31 return torch._shape_as_tensor(x)
Dsymbolic_opset9.py321 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)
/external/executorch/exir/
Dcontrol_flow.py91 proxy = torch.ops.aten._shape_as_tensor.default(x.proxy)
/external/pytorch/test/mobile/model_test/
Dtensor_ops.py96 a._shape_as_tensor(),
/external/pytorch/torch/jit/
D_shape_functions.py1086 def _shape_as_tensor(self: List[int]) -> List[int]: function
1420 "aten::_shape_as_tensor(Tensor self) -> Tensor", _shape_as_tensor
/external/pytorch/torch/csrc/jit/passes/onnx/
Dconstant_fold.cpp438 updated_val = at::_shape_as_tensor(inputTensorValues[0]); in runTorchBackendForOnnx()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py1835 shape_tensor = _shape_as_tensor(shape, row_partition_tensors[0].dtype)
2979 def _shape_as_tensor(shape, dtype): function
/external/pytorch/torch/_dynamo/variables/
Dtorch.py99 torch._shape_as_tensor,
/external/pytorch/aten/src/ATen/native/
DTensorShape.cpp366 Tensor _shape_as_tensor(const Tensor& self) { in _shape_as_tensor() function
/external/pytorch/test/dynamo/
Dtest_misc.py5151 return 1 + torch._shape_as_tensor(x)[0]