Searched defs:tensor_add (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_add_relu_fusion.cc | 33 VectorRef tensor_add = VectorRef({prim::kPrimAdd, tuple_get_item, z_}); in DefinePattern() local
43 auto tensor_add = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| D | post_batch_norm_add_relu_fusion.cc | 33 VectorRef tensor_add = VectorRef({prim::kPrimAdd, z_, tuple_get_item}); in DefinePattern() local
43 auto tensor_add = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| D | add_relu_grad_v2_fusion.cc | 72 auto tensor_add = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| D | add_relu_v2_fusion.cc | 70 auto tensor_add = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| /third_party/mindspore/tests/st/cpp/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/tests/st/pynative/parser/ |
| D | test_parser_construct.py | 79 def tensor_add(x, y): function
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ops/ |
| D | test_multitype.py | 28 tensor_add = P.Add() variable
|
| D | test_hypermap.py | 31 tensor_add = P.Add() variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | onehot_info.cc | 188 auto tensor_add = gen_g.PushBack({gen_g.NewOpInst(ADD), mul2, CreateInt32Tensor(1)}); in ComputeReplaceGraph() local
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | tensor_add_info_test.cc | 31 AddInfoPtr tensor_add, tensor_add1; variable
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 51 tensor_add = P.Add() variable
|