| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_compare.py | 82 def get_less_net_output(x0, y0, x1, y1, x2, y2, x3, y3, enable_graph_kernel=False): argument
92 def get_greater_net_output(x0, y0, x1, y1, x2, y2, x3, y3, enable_graph_kernel=False): argument
102 def get_less_equal_net_output(x0, y0, x1, y1, x2, y2, x3, y3, enable_graph_kernel=False): argument
112 def get_greater_equal_net_output(x0, y0, x1, y1, x2, y2, x3, y3, enable_graph_kernel=False): argument
|
| D | test_reorder_ops.py | 38 def get_castup_output(x0, enable_graph_kernel=False): argument
68 def get_castdown_output(x0, enable_graph_kernel=False): argument
|
| D | test_matmul.py | 41 def get_output(i0, i1, enable_graph_kernel=False): argument
47 def get_output1(i0, i1, i2, enable_graph_kernel=False): argument
|
| D | test_gelu.py | 48 def gelu(x, enable_graph_kernel=False): argument
69 def gelu_grad(input_dy, input_x, input_y, enable_graph_kernel=False): argument
|
| D | test_low_precision.py | 42 def get_output(x, y, z, net, enable_graph_kernel=False): argument
73 def get_output(x, y, net, enable_graph_kernel=False): argument
|
| D | test_recompute.py | 96 def get_output(i0, net, enable_graph_kernel=False): argument
114 def get_output(i0, i1, net, enable_graph_kernel=False): argument
132 def get_output(i0, i1, i2, net, enable_graph_kernel=False): argument
|
| D | test_identity.py | 31 def get_output(x, enable_graph_kernel=False): argument
|
| D | test_equal_count.py | 32 def get_output(*inp, enable_graph_kernel=False): argument
|
| D | test_argmax.py | 33 def get_output(x, axis, enable_graph_kernel=False): argument
|
| D | test_bias_add_grad.py | 33 def get_output(dout, enable_graph_kernel=False): argument
|
| D | test_layernorm.py | 44 def get_layernorm_output(x, gamma, beta, begin_norm_axis, begin_params_axis, enable_graph_kernel=Fa… argument
53 …rad_output(x, dy, var, mean, gamma, begin_norm_axis, begin_params_axis, enable_graph_kernel=False): argument
|
| D | test_square_sum_all.py | 31 def get_output(inp0, inp1, enable_graph_kernel=False): argument
|
| D | test_assign_add.py | 35 def get_output(x2, y2, enable_graph_kernel=False): argument
|
| D | test_relu.py | 32 def get_output(x, enable_graph_kernel=False): argument
|
| D | test_reduce_min.py | 33 def get_output(x, axis, keep_dims, enable_graph_kernel=False): argument
|
| D | test_softmax.py | 30 def get_output(x, enable_graph_kernel=False): argument
|
| D | test_cast_matmul_fusion.py | 37 def get_output(i0, i1, i2, enable_graph_kernel=False): argument
|
| D | test_slice.py | 33 def get_output(x, begin, size, enable_graph_kernel=False): argument
|
| D | test_reduce_max.py | 33 def get_output(x, axis, keep_dims, enable_graph_kernel=False): argument
|
| D | test_squeeze.py | 32 def get_output(x, axis=(), enable_graph_kernel=False): argument
|
| D | test_squared_difference.py | 32 def get_output(x, y, enable_graph_kernel=False): argument
|
| D | test_maximum_grad.py | 44 def get_maximum_grad_output(input_x, input_y, input_dout, enable_graph_kernel=False): argument
|
| D | test_fuse.py | 43 def get_output(i0, i1, i2, enable_graph_kernel=False): argument
|
| D | test_tile.py | 33 def get_output(x, multiples, enable_graph_kernel=False): argument
|
| D | test_expand_dims.py | 32 def get_output(x, axis, enable_graph_kernel=False): argument
|