Searched defs:scale_parameter (Results 1 – 2 of 2) sorted by relevance
/third_party/mindspore/mindspore/nn/optim/ |
D | adafactor.py | 30 def _get_lr(step, rms, learning_rate, relative_step, warmup_init, scale_parameter, eps): argument 69 weight_decay, scale_lr, scale_parameter, relative_step, argument 296 scale_parameter=True, argument
|
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/ |
D | nnacl_fp32_serializer.cc | 85 void NNaclFp32Serializer::CodeStruct(const std::string &name, const ScaleParameter &scale_parameter… in CodeStruct()
|