Searched defs:assign_v (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | adam_weight_decay_fusion.cc | 124 VectorRef assign_v = VectorRef({prim::kPrimAssign, v_, next_v, next_state}); in DefinePattern() local
|
| D | adam_fusion.cc | 122 VectorRef assign_v = VectorRef({prim::kPrimAssign, v_, next_v, next_state}); in DefinePattern() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | adam_weight_decay_unify_mindir.cc | 185 …auto assign_v = CreateAssignCNode(func_graph, input_list[kIndex3], new_cnode_outputs[kIndex0], inp… in CreateAdamApplyOneWithDecay() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | encoder_layer_fusion.cc | 200 auto assign_v = VectorRef({is_assign_v, v_past_, mul_v, var_assign_v}); in DefinePatternInitReset() local
394 auto assign_v = VectorRef({is_assign_v, v_past_, DefineBatchValidLength(tuple_v), var_assign_v}); in DefineDependKV() local
|