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

Example 36 with PropertyMapping

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

the class Stack method mappingsForFunction.

@Override
public Map<String, Map<String, PropertyMapping>> mappingsForFunction() {
    Map<String, Map<String, PropertyMapping>> ret = new HashMap<>();
    Map<String, PropertyMapping> map = new HashMap<>();
    val axisMapping = PropertyMapping.builder().onnxAttrName("axis").tfAttrName("axis").propertyNames(new String[] { "axis" }).build();
    map.put("axis", axisMapping);
    for (val name : tensorflowNames()) ret.put(name, map);
    return ret;
}