Searched refs:ge_tensor_desc (Results 1 – 1 of 1) sorted by relevance
100 ge::GeTensorDesc ge_tensor_desc; in ConvertInputShapeAndType() local101 ge_tensor_desc.SetFormat(ge_format); in ConvertInputShapeAndType()102 ge_tensor_desc.SetDataType(ge_dtype); in ConvertInputShapeAndType()103 ge_tensor_desc.SetShape(ge::GeShape(ge_shape)); in ConvertInputShapeAndType()104 ge_tensor_desc.SetOriginShape(ge::GeShape(ge_ori_shape)); in ConvertInputShapeAndType()105 ge_tensor_desc.SetName(input_name); in ConvertInputShapeAndType()106 (void)(*op_desc)->AddInputDesc(input_name, ge_tensor_desc); in ConvertInputShapeAndType()129 ge::GeTensorDesc ge_tensor_desc; in ConvertOutputShapeAndType() local130 ge_tensor_desc.SetFormat(ge_format); in ConvertOutputShapeAndType()131 ge_tensor_desc.SetDataType(ge_dtype); in ConvertOutputShapeAndType()[all …]