Searched refs:_shape_as_tensor (Results 1 – 10 of 10) sorted by relevance
31 return torch._shape_as_tensor(x)
321 def _shape_as_tensor(g: jit_utils.GraphContext, input): function5393 shape_tensor = _shape_as_tensor(g, self)5425 cum_adv_index_shape_tensor = _shape_as_tensor(g, cum_adv_index)
91 proxy = torch.ops.aten._shape_as_tensor.default(x.proxy)
96 a._shape_as_tensor(),
1086 def _shape_as_tensor(self: List[int]) -> List[int]: function1420 "aten::_shape_as_tensor(Tensor self) -> Tensor", _shape_as_tensor
438 updated_val = at::_shape_as_tensor(inputTensorValues[0]); in runTorchBackendForOnnx()
1835 shape_tensor = _shape_as_tensor(shape, row_partition_tensors[0].dtype)2979 def _shape_as_tensor(shape, dtype): function
99 torch._shape_as_tensor,
366 Tensor _shape_as_tensor(const Tensor& self) { in _shape_as_tensor() function
5151 return 1 + torch._shape_as_tensor(x)[0]