List of all items
Structs
- AdadeltaOp
- AdagradOp
- AdamOp
- AlternateLearningRate
- AtomicIterOp
- AtomicIterOpStats
- ClipTensorByScalingOp
- CompositeCosineLearningRate
- CompositeCyclicalLearningRate
- CompositeLearningRate
- CompositeLearningRateItem
- ConstantThenLinearWarmupLearningRate
- ConstantWarmupLearningRate
- CosineLearningRate
- CyclicalLearningRate
- ExpLearningRate
- FP16MomentumSGDUpdateOp
- FP32MomentumSGDUpdateOp
- FixedLearningRate
- FtrlOp
- FtrlParams
- GFtrlOp
- GFtrlParams
- GateLearningRate
- HillLearningRate
- InvLearningRate
- IterOp
- LarsOp
- LearningRateAdaptionOp
- LearningRateOp
- LinearWarmupLearningRate
- MomentumSGDOp
- MomentumSGDUpdateOp
- MutexDeserializer
- MutexSerializer
- PieceWarmupLearningRate
- PolyLearningRate
- RmsPropOp
- RowWiseCounterOp
- RowWiseSparseAdagradFusedWithSparseLengthsSumGradientOp
- RowWiseSparseAdagradFusedWithSparseLengthsWeightedSumGradientApproxOp
- RowWiseSparseAdagradFusedWithSparseLengthsWeightedSumGradientOp
- RowWiseSparseAdagradOp
- RowWiseSparseAdamOp
- SlopeLearningRate
- SparseAdadeltaOp
- SparseAdagradFusedWithSparseLengthsSumGradientOp
- SparseAdagradFusedWithSparseLengthsWeightedSumGradientApproxOp
- SparseAdagradFusedWithSparseLengthsWeightedSumGradientOp
- SparseAdagradOp
- SparseAdamOp
- SparseFtrlOp
- SparseMomentumSGDUpdateOp
- SparseStormOp
- SparseWngradOp
- StepLearningRate
- StormOp
- WeightScaleOp
- WngradOp
- YellowFinOp
- adagrad_update_prefetch_inlined
Traits
Functions
- adadelta_update
- adagrad_update
- adagrad_update_output_effective_lr
- adagrad_update_output_effective_lr_and_update
- adam_compute
- adam_compute_output_grad
- adam_update
- compute_square_average_inlined
- compute_square_average_with_weight_decay_inlined
- compute_square_average_with_weight_decay_inlined_f16_weights
- cost_inference_for_adagrad
- cost_inference_for_row_wise_sparse_adagrad
- cost_inference_for_sparse_adagrad
- dot
- dot_float_half_float
- dot_float_x3
- fp16_momentum_sgd_update
- fp32_momentum_sgd_update
- ftrl_compute
- ftrl_update
- gftrl_compute
- gftrl_update
- increment_iter
- lr_update
- momentum_sgd_update
- radam_compute
- radam_compute_output_grad
- radam_update
- sgn
- storm_update
- weight_scale_update
- weight_scale_update_kernel
- wngrad_update
- wngrad_update_output_effective_lr
- wngrad_update_output_effective_lr_and_update