Search in sources :

Example 1 with BooleanAdapter

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

the class CumSum method attributeAdaptersForFunction.

@Override
public Map<String, Map<String, AttributeAdapter>> attributeAdaptersForFunction() {
    Map<String, Map<String, AttributeAdapter>> ret = new HashMap<>();
    Map<String, AttributeAdapter> tfMappings = new LinkedHashMap<>();
    tfMappings.put("exclusive", new BooleanAdapter());
    tfMappings.put("reverse", new BooleanAdapter());
    ret.put(tensorflowName(), tfMappings);
    return ret;
}
Also used : AttributeAdapter(org.nd4j.imports.descriptors.properties.AttributeAdapter) BooleanAdapter(org.nd4j.imports.descriptors.properties.adapters.BooleanAdapter)

Example 2 with BooleanAdapter

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

the class CumProd method attributeAdaptersForFunction.

@Override
public Map<String, Map<String, AttributeAdapter>> attributeAdaptersForFunction() {
    Map<String, Map<String, AttributeAdapter>> ret = new HashMap<>();
    Map<String, AttributeAdapter> tfMappings = new LinkedHashMap<>();
    tfMappings.put("exclusive", new BooleanAdapter());
    tfMappings.put("reverse", new BooleanAdapter());
    ret.put(tensorflowName(), tfMappings);
    return ret;
}
Also used : AttributeAdapter(org.nd4j.imports.descriptors.properties.AttributeAdapter) BooleanAdapter(org.nd4j.imports.descriptors.properties.adapters.BooleanAdapter)

Aggregations

AttributeAdapter (org.nd4j.imports.descriptors.properties.AttributeAdapter)2 BooleanAdapter (org.nd4j.imports.descriptors.properties.adapters.BooleanAdapter)2