Home
last modified time | relevance | path

Searched defs:onnx_tensor_type (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport.cpp612 onnx::TypeProto_Tensor* onnx_tensor_type, in TensorTypeToONNXType()
657 onnx::TypeProto_Tensor* onnx_tensor_type = in EncodeValueInfoType() local
674 onnx::TypeProto_Tensor* onnx_tensor_type = onnx_type->mutable_tensor_type(); in EncodeValueInfoType() local
677 onnx::TypeProto_Tensor* onnx_tensor_type = onnx_type->mutable_tensor_type(); in EncodeValueInfoType() local
680 onnx::TypeProto_Tensor* onnx_tensor_type = onnx_type->mutable_tensor_type(); in EncodeValueInfoType() local
686 onnx::TypeProto* onnx_tensor_type = sequence_type->mutable_elem_type(); in EncodeValueInfoType() local
693 onnx::TypeProto_Tensor* onnx_tensor_type = in EncodeValueInfoType() local
708 onnx::TypeProto_Tensor* onnx_tensor_type = in EncodeValueInfoType() local
1269 onnx::TypeProto_Tensor* onnx_tensor_type = in EncodeTypeProto() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp120 const onnx::TypeProto_Tensor& onnx_tensor_type, in TorchTensorTypeFromONNX()
162 const auto& onnx_tensor_type = onnx_seq_elem_type.tensor_type(); in TorchListTypeFromONNX() local