Searched defs:tensor_add (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | post_batch_norm_add_relu_fusion.cc | 41 VectorRef tensor_add = VectorRef({prim::kPrimAdd, z_, tuple_get_item}); in DefinePattern() local
51 auto tensor_add = common::AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| D | batch_norm_add_relu_fusion.cc | 42 VectorRef tensor_add = VectorRef({prim::kPrimAdd, tuple_get_item, z_}); in DefinePattern() local
79 auto tensor_add = common::AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/fallback/ |
| D | utils.py | 25 tensor_add = ops.Add() variable
|
| /third_party/mindspore/mindspore-src/source/tests/st/cpp/cxx_api/model/ |
| D | test_tensor_add.cc | 39 Model tensor_add; in TEST_F() local
|
| D | test_dynamic_batch_size.cc | 53 Model tensor_add; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/pynative_mode/ops/ |
| D | test_multitype.py | 27 tensor_add = P.Add() variable
|
| D | test_hypermap.py | 29 tensor_add = P.Add() variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | add_rms_norm_fusion.cc | 85 auto tensor_add = common::AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| D | add_cast_rms_norm_cast_fusion.cc | 108 auto tensor_add = common::AnfAlgo::GetInputNode(utils::cast<CNodePtr>(cast_node), 0); in Process() local
|
| D | add_layer_norm_fusion.cc | 141 auto tensor_add = common::AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | onehot_info.cc | 198 auto tensor_add = gen_g.PushBack({gen_g.NewOpInst(ADD), mul2, CreateInt32Tensor(1)}); in ComputeReplaceGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | add_layernorm_fusion.cc | 319 auto tensor_add = cnode->input(kInputIndex1); in Process() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/ops_info/ |
| D | tensor_add_info_test.cc | 31 AddInfoPtr tensor_add, tensor_add1; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | math_func.py | 115 tensor_add = P.Add() variable
|