code: public long getTotalMemoryBytes(int minibatchSize, @NonNull MemoryUseMode memoryUseMode,
                      @NonNull CacheMode cacheMode) {
          return getTotalMemoryBytes(minibatchSize, memoryUseMode, cacheMode, DataTypeUtil.getDtypeFromContext());
      }
docs: Get the total memory use in bytes for the given configuration (using the current ND4J data type)
  
  @param minibatchSize Mini batch size to estimate the memory for
  @param memoryUseMode The memory use mode (training or inference)
  @param cacheMode     The CacheMode to use
  @return The estimated total memory consumption in bytes