Home
last modified time | relevance | path

Searched full:_reshape_from_tensor (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset12.py427 output_a = opset9._reshape_from_tensor(g, new_input_a, shape_sizes)
437 output_a = opset9._reshape_from_tensor(g, new_input_a, shape_sizes)
450 output_b = opset9._reshape_from_tensor(g, new_input_b, shape_sizes)
460 output_b = opset9._reshape_from_tensor(g, new_input_b, shape_sizes)
465 return opset9._reshape_from_tensor(g, output, shape_sizes)
H A Doperators.py47 return torch._reshape_from_tensor(x, shape)
H A Dsymbolic_helper.py1475 from torch.onnx.symbolic_opset9 import _reshape_from_tensor
1477 return _reshape_from_tensor(g, input, final_shape)
H A Dsymbolic_opset9.py325 @_onnx_symbolic("aten::_reshape_from_tensor")
326 def _reshape_from_tensor(g: jit_utils.GraphContext, input, shape): function
5668 t_reshaped = _reshape_from_tensor(g, t, g.op("Concat", *shape_i, axis_i=0))
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dmodel_ops.yaml40 aten::_reshape_from_tensor: 16
H A Dcoverage.yaml1057 aten::_reshape_from_tensor: 10
/aosp_15_r20/external/pytorch/test/
H A Dallowlist_for_publicAPI.json1283 "_reshape_from_tensor",
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py1544 "torch._reshape_from_tensor",
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp356 Tensor _reshape_from_tensor(const Tensor& self, const Tensor& shape_tensor) { in _reshape_from_tensor() function
H A Dnative_functions.yaml307 - func: _reshape_from_tensor(Tensor self, Tensor shape) -> Tensor