Examples with DataTypeAdapter org.nd4j.imports.descriptors.properties.adapters.DataTypeAdapter used on opensource projects

Example 1 with DataTypeAdapter

use of org.nd4j.imports.descriptors.properties.adapters.DataTypeAdapter in project nd4j by deeplearning4j.

the class Cast method attributeAdaptersForFunction.

@Override
public Map<String, Map<String, AttributeAdapter>> attributeAdaptersForFunction() {
    Map<String, Map<String, AttributeAdapter>> ret = new LinkedHashMap<>();
    Map<String, AttributeAdapter> tfAdapters = new LinkedHashMap<>();
    val fields = DifferentialFunctionClassHolder.getInstance().getFieldsForFunction(this);
    tfAdapters.put("typeDst", new DataTypeAdapter());
    ret.put(tensorflowName(), tfAdapters);
    return ret;
}