Searched refs:input_dout (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_maximum_grad.py | 40 input_dout = Tensor(input_dout_np)
41 return input_x, input_y, input_dout
44 def get_maximum_grad_output(input_x, input_y, input_dout, enable_graph_kernel=False): argument
47 result = net(input_x, input_y, input_dout)
52 input_x, input_y, input_dout = gen_data()
53 result_off = get_maximum_grad_output(input_x, input_y, input_dout, False)
54 result_on = get_maximum_grad_output(input_x, input_y, input_dout, True)
|
| D | test_minimum_grad.py | 40 input_dout = Tensor(input_dout_np)
41 return input_x, input_y, input_dout
44 def get_minimum_grad_output(input_x, input_y, input_dout, enable_graph_kernel=False): argument
47 result = net(input_x, input_y, input_dout)
52 input_x, input_y, input_dout = gen_data()
53 result_off = get_minimum_grad_output(input_x, input_y, input_dout, False)
54 result_on = get_minimum_grad_output(input_x, input_y, input_dout, True)
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_maximum_grad.py | 42 input_dout = np.maximum(input_x, input_y).astype(np.float32)
44 result = net(Tensor(input_x), Tensor(input_y), Tensor(input_dout))
45 dx = input_dout * (input_x >= input_y)
46 dy = input_dout - dx
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | maximum_grad.py | 31 input_x, input_y, input_dout = self.inputs
34 dx = graph_builder.emit('Mul', [ge_result, input_dout])
35 dy = graph_builder.emit('Sub', [input_dout, dx])
|
| D | minimum_grad.py | 30 input_x, input_y, input_dout = self.inputs
34 dx = graph_builder.emit('Mul', [le_result, input_dout])
35 dy = graph_builder.emit('Sub', [input_dout, dx])
|