cavis/cavis-nd4j/cavis-nd4j-tensorflow/build.gradle

/*
 *
 *    ******************************************************************************
 *    *
 *    * This program and the accompanying materials are made available under the
 *    * terms of the Apache License, Version 2.0 which is available at
 *    * https://www.apache.org/licenses/LICENSE-2.0.
 *    *
 *    *  See the NOTICE file distributed with this work for additional
 *    *  information regarding copyright ownership.
 *    * Unless required by applicable law or agreed to in writing, software
 *    * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 *    * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 *    * License for the specific language governing permissions and limitations
 *    * under the License.
 *    *
 *    * SPDX-License-Identifier: Apache-2.0
 *    *****************************************************************************
 *
 */
ext {
    buildTarget = rootProject.ext.buildTarget
}

dependencies {
    implementation projects.cavisDnn.cavisDnnApi
    implementation projects.cavisNative.cavisNativeBlas
    implementation group: "org.bytedeco", name: "tensorflow"
    testRuntimeOnly group: "org.bytedeco", name: "tensorflow", classifier: buildTarget
    if(buildTarget.contains("windows") || buildTarget.contains("linux")) {
        testRuntimeOnly group: "org.bytedeco", name: 'tensorflow', classifier: "${buildTarget}-gpu", version: ''
    }
    implementation "commons-io:commons-io"
    implementation "com.google.code.gson:gson"
    implementation "com.google.protobuf:protobuf-java"
    implementation "com.google.protobuf:protobuf-java-util"
    implementation "org.slf4j:slf4j-api"
}