Searched defs:onnx_type (Results 1 – 7 of 7) sorted by relevance
38 int64_t onnx_type = -1; in ScalarTypeToONNXType() local300 const int64_t onnx_type = ScalarTypeToONNXType(scalar_type); in UpdateScalarTypeForInputs() local368 const int64_t onnx_type = ScalarTypeToONNXType(scalar_type); in RecoverScalarTypeForOutput() local
410 auto onnx_type = ATenTypeToOnnxType(scalar_type); in InferShapeTypeForUninitializedOutput() local415 auto onnx_type = ATenTypeToOnnxType(scalar_type); in InferShapeTypeForUninitializedOutput() local
63 std::optional<at::ScalarType> ONNXTypeToATenType(int32_t onnx_type) { in ONNXTypeToATenType()
923 auto onnx_type = castNode->i(attr::to); in fuseLogSoftmaxNllLoss() local
1659 auto onnx_type = ATenTypeToOnnxType(scalar_type); in SpecialPostProcess() local
257 def onnx_type(self) -> _C_onnx.TensorProtoDataType: member in JitScalarType
644 onnx::TypeProto* onnx_type, in EncodeValueInfoType()