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;
}
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;
}
Aggregations