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

38 lines
1.6 KiB
Groovy

/*
*
* ******************************************************************************
* *
* * 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"
}