/** * <pre> * The default configuration for sessions that run on this server. * </pre> * * <code>.tensorflow.ConfigProto default_session_config = 4;</code> */ public org.tensorflow.framework.ConfigProto getDefaultSessionConfig() { return defaultSessionConfig_ == null ? org.tensorflow.framework.ConfigProto.getDefaultInstance() : defaultSessionConfig_; } /**
public org.tensorflow.framework.ConfigProto getDefaultInstanceForType() { return org.tensorflow.framework.ConfigProto.getDefaultInstance(); }
/** * <pre> * The default configuration for sessions that run on this server. * </pre> * * <code>.tensorflow.ConfigProto default_session_config = 4;</code> */ public org.tensorflow.framework.ConfigProto getDefaultSessionConfig() { if (defaultSessionConfigBuilder_ == null) { return defaultSessionConfig_ == null ? org.tensorflow.framework.ConfigProto.getDefaultInstance() : defaultSessionConfig_; } else { return defaultSessionConfigBuilder_.getMessage(); } } /**
/** * <pre> * The default configuration for sessions that run on this server. * </pre> * * <code>.tensorflow.ConfigProto default_session_config = 4;</code> */ public org.tensorflow.framework.ConfigProtoOrBuilder getDefaultSessionConfigOrBuilder() { if (defaultSessionConfigBuilder_ != null) { return defaultSessionConfigBuilder_.getMessageOrBuilder(); } else { return defaultSessionConfig_ == null ? org.tensorflow.framework.ConfigProto.getDefaultInstance() : defaultSessionConfig_; } } /**
public static org.tensorflow.framework.ConfigProto getAlignedWithNd4j() { org.tensorflow.framework.ConfigProto configProto = org.tensorflow.framework.ConfigProto.getDefaultInstance(); ConfigProto.Builder builder1 = configProto.toBuilder().addDeviceFilters(TensorflowConversion.defaultDeviceForThread()); try { //cuda if(Nd4j.getBackend().getClass().getName().toLowerCase().contains("jcu")) { builder1.setGpuOptions(GPUOptions.newBuilder() .setAllowGrowth(true) .setPerProcessGpuMemoryFraction(0.5) .build()); } //cpu else { } } catch (Exception e) { e.printStackTrace(); } return builder1.build(); }
public Builder mergeFrom(org.tensorflow.framework.ConfigProto other) { if (other == org.tensorflow.framework.ConfigProto.getDefaultInstance()) return this; internalGetMutableDeviceCount().mergeFrom( other.internalGetDeviceCount());