Home
last modified time | relevance | path

Searched refs:c_weight (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_lstm_cell_fusion.cc401 auto c_weight = func_graph->add_parameter(); in CreateLSTMNode() local
402 MS_CHECK_TRUE_RET(c_weight != nullptr, nullptr); in CreateLSTMNode()
403 c_weight->set_name(base_name + "_weight_c"); in CreateLSTMNode()
405 c_weight->set_abstract(weight->abstract()->Clone()); in CreateLSTMNode()
408 if (SplitWeights(weight, i_weight, c_weight, static_cast<int>(hidden_shape.back())) != RET_OK) { in CreateLSTMNode()
433 …std::vector<AnfNodePtr> new_node_inputs = {value_node, input_tensor_node, i_weight, c_weight, bias… in CreateLSTMNode()
Dtflite_lstm_cell_fusion.cc576 auto c_weight = func_graph->add_parameter(); in CreateLSTMNode() local
577 MS_CHECK_TRUE_RET(c_weight != nullptr, nullptr); in CreateLSTMNode()
578 status = GetConcatedParam(c_weights, c_weight, false); in CreateLSTMNode()
582 c_weight->set_name(base_name + "_weight_c"); in CreateLSTMNode()
600 …std::vector<AnfNodePtr> new_node_inputs = {value_node, input_tensor_node, i_weight, c_weight, bias… in CreateLSTMNode()
/third_party/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_sample_soa.c1442 LLVMValueRef w00, w01, w10, w11, wx0, wy0, c_weight, tmp; in lp_build_sample_image_linear() local
1451 c_weight = lp_build_select(coord_bld, c00, w00, coord_bld->zero); in lp_build_sample_image_linear()
1452 c_weight = lp_build_select(coord_bld, c01, w01, c_weight); in lp_build_sample_image_linear()
1453 c_weight = lp_build_select(coord_bld, c10, w10, c_weight); in lp_build_sample_image_linear()
1454 c_weight = lp_build_select(coord_bld, c11, w11, c_weight); in lp_build_sample_image_linear()
1460 c_weight = lp_build_mul(coord_bld, c_weight, one_third); in lp_build_sample_image_linear()
1461 w00 = lp_build_add(coord_bld, w00, c_weight); in lp_build_sample_image_linear()
1463 w01 = lp_build_add(coord_bld, w01, c_weight); in lp_build_sample_image_linear()
1465 w10 = lp_build_add(coord_bld, w10, c_weight); in lp_build_sample_image_linear()
1467 w11 = lp_build_add(coord_bld, w11, c_weight); in lp_build_sample_image_linear()