• Home
• com.facebook
• diffkt
• 0.0.1-DEV2
• source code
• TrainableLayer.kt

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.diffkt.model.TrainableLayer.kt Maven / Gradle / Ivy

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

package org.diffkt.model

interface TrainableLayer> : TrainableComponent, Layer {
    override fun cpu(): T = withTrainables(trainables.map { it.cpu() })
    override fun gpu(): T = withTrainables(trainables.map { it.gpu() })
}

interface TrainableLayerSingleInput> : TrainableLayer, TrainableComponent, LayerSingleInput