| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_relu_op.py | 29 self.relu = P.ReLU()
32 return self.relu(x)
39 self.relu = P.ReLU()
43 return self.relu(x_conv)
58 relu = NetRelu()
59 output = relu(x)
63 relu = NetRelu()
64 output = relu(x)
80 relu = NetRelu()
81 output = relu(x)
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | derelu_fusion.cc | 40 CNodePtr CreateReluV2(const FuncGraphPtr &graph, const CNodePtr &relu) { in CreateReluV2() argument
42 MS_EXCEPTION_IF_NULL(relu); in CreateReluV2()
43 CheckCNodeInputSize(relu, kReluInputTensorNum); in CreateReluV2()
46 std::vector<AnfNodePtr> inputs = {NewValueNode(prim), relu->input(kIndex1)}; in CreateReluV2()
49 new_node->set_scope(relu->scope()); in CreateReluV2()
53 if (AnfAlgo::IsDynamicShape(relu)) { in CreateReluV2()
56 std::vector<size_t> mask_shape = AnfAlgo::GetOutputInferShape(relu, 0); in CreateReluV2()
61 auto input_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(relu, 0); in CreateReluV2()
77 auto types = {AnfAlgo::GetOutputInferDataType(relu, 0), mask_dtype}; in CreateReluV2()
78 auto shapes = {AnfAlgo::GetOutputInferShape(relu, 0), mask_shape}; in CreateReluV2()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | relu_v2_pass.cc | 67 CNodePtr CreateReluV2(const FuncGraphPtr &graph, const CNodePtr &relu) { in CreateReluV2() argument
69 MS_EXCEPTION_IF_NULL(relu); in CreateReluV2()
70 CheckCNodeInputSize(relu, kReluInputTensorNum); in CreateReluV2()
73 std::vector<AnfNodePtr> inputs = {NewValueNode(prim), relu->input(1)}; in CreateReluV2()
76 new_node->set_scope(relu->scope()); in CreateReluV2()
78 if (AnfAlgo::IsDynamicShape(relu)) { in CreateReluV2()
81 std::vector<size_t> output_shape = AnfAlgo::GetOutputInferShape(relu, 0); in CreateReluV2()
86 auto shapes = {AnfAlgo::GetOutputInferShape(relu, 0), mask_shape}; in CreateReluV2()
87 auto types = {AnfAlgo::GetOutputInferDataType(relu, 0), kNumberTypeUInt32}; in CreateReluV2()
135 auto relu = GetRelu(relu_grad); in Process() local
[all …]
|
| /third_party/mindspore/tests/st/networks/models/ |
| D | alexnet.py | 28 self.relu = P.ReLU()
37 x = self.relu(x)
40 x = self.relu(x)
43 x = self.relu(x)
45 x = self.relu(x)
47 x = self.relu(x)
51 x = self.relu(x)
53 x = self.relu(x)
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_tbe_reduce_eltwise_fusion.py | 31 self.relu = P.ReLU()
35 x = self.relu(x)
36 x = self.relu(x)
37 x = self.relu(x)
39 x = self.relu(x)
40 x = self.relu(x)
41 x = self.relu(x)
|
| D | test_tbe_eltwise_fusion_2.py | 31 self.relu = P.ReLU()
35 x = self.relu(x)
36 x = self.relu(x)
37 x = self.relu(x)
39 x = self.relu(x)
40 x = self.relu(x)
41 x = self.relu(x)
|
| /third_party/mindspore/tests/st/dynamic_shape/ |
| D | test_getnext_dynamic_pipeline.py | 56 self.relu = P.ReLU()
59 x1 = self.relu(x1)
60 x1 = self.relu(x1)
62 x2 = self.relu(x2)
64 x3 = self.relu(x3)
65 x3 = self.relu(x3)
67 x4 = self.relu(x4)
69 x5 = self.relu(x5)
|
| /third_party/mindspore/tests/ut/python/pipeline/parse/ |
| D | test_operator.py | 32 self.relu = ReLU()
40 x = self.relu(x)
44 x = self.relu(x)
48 x = self.relu(x)
52 x = self.relu(x)
56 x = self.relu(x)
60 x = self.relu(x)
64 x = self.relu(x)
67 x = self.relu(x)
79 self.relu = ReLU()
[all …]
|
| D | test_fix_bug.py | 33 self.relu = nn.ReLU()
38 x = self.relu(x)
48 self.relu = nn.ReLU()
53 x = self.relu(x)
84 self.relu = nn.ReLU()
105 self.relu = nn.ReLU()
113 x = self.relu(x)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | common_func_fp32.h | 34 size_t relu; member
53 … size_t in_sw_step, size_t in_kh_step, size_t in_kw_step, size_t relu, size_t relu6);
58 … size_t in_kh_step, size_t in_kw_step, size_t kernel_w, size_t relu, size_t relu6);
77 … size_t in_sh_step, size_t in_sw_step, size_t in_kh_step, size_t in_kw_step, size_t relu,
81 … int row_size, int channel, int output_h, int output_w, size_t relu, size_t relu6);
84 … int row_size, int channel, int output_h, int output_w, size_t relu, size_t relu6);
87 int in_kw_step, int channel, size_t relu, size_t relu6);
90 int in_kw_step, int channel, size_t relu, size_t relu6);
93 int in_kw_step, int channel, size_t relu, size_t relu6);
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_alexnet.py | 41 self.relu = nn.ReLU()
50 x = self.relu(x)
53 x = self.relu(x)
56 x = self.relu(x)
58 x = self.relu(x)
60 x = self.relu(x)
64 x = self.relu(x)
66 x = self.relu(x)
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | insert_tensor_move_for_hccl_op.py | 28 relu = P.ReLU() variable
47 res1 = relu(x)
53 res1 = relu(x)
59 res1 = relu(x)
108 x = relu(a)
115 x = relu(a)
129 x = relu(a)
136 x = relu(a)
|
| /third_party/mindspore/tests/st/hook/ |
| D | test_hook_function.py | 33 self.relu = nn.ReLU()
40 x = self.relu(x)
47 self.relu = nn.ReLU()
55 x = self.relu(x)
102 self.relu = nn.ReLU()
103 self.relu.register_backward_hook(cell_hook_function)
108 x = self.relu(x)
115 self.relu = nn.ReLU()
116 self.relu.register_backward_hook(cell_hook_function)
122 x = self.relu(x)
|
| /third_party/mindspore/tests/ut/python/pynative_mode/nn/ |
| D | test_activation.py | 24 relu = nn.ReLU()
26 output = relu.construct(input_data)
32 relu = nn.get_activation('relu')
35 output = relu.construct(input_data)
41 relu = nn.get_activation('relu')
44 output = relu.construct(input_data)
|
| /third_party/mindspore/tests/st/pynative/parser/ |
| D | test_parser_tensor_assign.py | 36 self.relu = ReLU()
40 x = self.relu(x)
58 self.relu = ReLU()
62 x = self.relu(x)
84 self.relu = ReLU()
88 x = self.relu(x)
108 self.relu = ReLU()
113 x = self.relu(x)
|
| /third_party/mindspore/tests/ut/python/utils/ |
| D | test_initializer_fuzz.py | 39 self.relu = nn.ReLU()
46 x = self.relu(x)
80 self.relu = nn.ReLU()
85 x = self.max_pool2d(self.relu(self.conv1(x)))
86 x = self.max_pool2d(self.relu(self.conv2(x)))
88 x = self.relu(self.fc1(x))
89 x = self.relu(self.fc2(x))
|
| D | test_checkparam.py | 35 self.relu = nn.ReLU()
40 x = self.max_pool2d(self.relu(self.conv1(x)))
41 x = self.max_pool2d(self.relu(self.conv2(x)))
43 x = self.relu(self.fc1(x))
44 x = self.relu(self.fc2(x))
|
| /third_party/mindspore/tests/st/ops/ascend/ |
| D | test_relu.py | 29 self.relu = P.ReLU(strategy=None)
33 return self.relu(x)
38 relu = Net()
39 output = relu(Tensor(x))
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | model.py | 55 self.relu = nn.ReLU()
61 x = self.relu(x)
64 x = self.relu(x)
68 x = self.relu(x)
70 x = self.relu(x)
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | depthwiseconv_eltwise_fusion_pass.cc | 42 auto relu = cnode->input(kIndex1); in MatchDepthwiseConvRelu() local
43 MS_EXCEPTION_IF_NULL(relu); in MatchDepthwiseConvRelu()
44 …if (cnode->isa<CNode>() && (IsPrimitiveCNode(relu, prim::kPrimRelu) || IsPrimitiveCNode(relu, prim… in MatchDepthwiseConvRelu()
45 std::unordered_set<AnfNodePtr> record{cnode, relu}; in MatchDepthwiseConvRelu()
|
| /third_party/mindspore/tests/perf_test/ |
| D | lenet.py | 61 self.relu = nn.ReLU()
67 x = self.relu(x)
70 x = self.relu(x)
74 x = self.relu(x)
76 x = self.relu(x)
|
| /third_party/mindspore/tests/ut/python/model/ |
| D | test_lenet_core_after_exception.py | 36 self.relu = nn.ReLU()
41 x = self.max_pool2d(self.relu(self.conv1(x)))
42 x = self.max_pool2d(self.relu(self.conv2(x)))
44 x = self.relu(self.fc1(x))
45 x = self.relu(self.fc2(x))
|
| D | test_lenet.py | 38 self.relu = nn.ReLU()
43 x = self.max_pool2d(self.relu(self.conv1(x)))
44 x = self.max_pool2d(self.relu(self.conv2(x)))
46 x = self.relu(self.fc1(x))
47 x = self.relu(self.fc2(x))
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_relu.py | 29 self.relu = P.ReLU()
33 return self.relu(x)
39 relu = Net()
40 output = relu(Tensor(x))
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_relu_op.py | 32 self.relu = P.ReLU()
38 return self.relu(self.x)
45 relu = NetRelu()
46 output = relu()
|