Skip to content

Commit d3d8535

Browse files
committed
Add mapping from TypeReference to TensorFlow signatures.
1 parent 58ea3b5 commit d3d8535

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

com.ibm.wala.cast.python.ml/source/com/ibm/wala/cast/python/ml/types/TensorFlowTypes.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,10 @@ public boolean canConvertTo(DType other) {
187187
TypeName.string2TypeName("Ltensorflow/functions/ragged_constant")),
188188
AstMethodReference.fnSelector);
189189

190+
/** A mapping from a {@link TypeReference} to its associated TensorFlow signature. */
191+
public static final Map<TypeReference, String> TYPE_REFERENCE_TO_SIGNATURE =
192+
Map.of(RAGGED_CONSTANT.getDeclaringClass(), "tf.ragged.constant");
193+
190194
/**
191195
* Represents the TensorFlow float32 data type.
192196
*

0 commit comments

Comments
 (0)