tensorflow::ops

Function load_tpuembedding_adamparameters_grad_accum_debug

Source
pub fn load_tpuembedding_adamparameters_grad_accum_debug<O0: Into<Output>, O1: Into<Output>, O2: Into<Output>, O3: Into<Output>>(
    parameters: O0,
    momenta: O1,
    velocities: O2,
    gradient_accumulators: O3,
    scope: &mut Scope,
) -> Result<Operation>
Expand description

Shorthand for LoadTPUEmbeddingADAMParametersGradAccumDebug::new().build(parameters, momenta, velocities, gradient_accumulators, scope).