Searched defs:onnx_type (Results 1 – 2 of 2) sorted by relevance
163 TypeId OnnxNodeParser::GetDataTypeFromOnnx(onnx::TensorProto_DataType onnx_type) { in GetDataTypeFromOnnx()
866 OpNameInfo &set_onnx_type(const std::string &onnx_type) { in set_onnx_type()871 const std::string &onnx_type() const { return onnx_type_; } in onnx_type() function in mindspore::OpNameInfo2290 auto onnx_type = GetOutputType(node->input(kOneNum)); in ExportPrimGeLU() local3727 auto onnx_type = GetOutputType(node->input(kOneNum)); in ExportPrimFloorDiv() local3755 auto onnx_type = GetOutputType(node->input(kOneNum)); in ExportPrimFloorMod() local4271 auto onnx_type = GetOutputType(batch_norm_node->input(kOneNum)); in ExportMergeBatchNorm() local4332 auto onnx_type = GetOutputType(LayerNormNode->input(kOneNum)); in ExportMergeLayerNorm() local