| /third_party/mindspore/mindspore-src/source/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_fast_gelu.py | 49 def FastGeLU(x, enable_graph_kernel=False): argument
65 def gelu_grad(input_dy, input_x, input_y, enable_graph_kernel=False): argument
|
| D | test_gelu.py | 49 def gelu(x, enable_graph_kernel=False): argument
65 def gelu_grad(input_dy, input_x, input_y, enable_graph_kernel=False): argument
|
| D | test_matmul.py | 42 def get_output(i0, i1, enable_graph_kernel=False): argument
48 def get_output1(i0, i1, i2, 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 | gk_utils.py | 21 def __init__(self, enable_graph_kernel=True): argument
|
| D | test_easy_fuse_cpu.py | 41 def get_output(i0, i1, enable_graph_kernel=False): argument
|
| D | test_apply_momentum.py | 33 def get_output(lr, grad, moment, enable_graph_kernel=False): argument
|
| D | test_identity.py | 31 def get_output(x, 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_argmax.py | 33 def get_output(x, axis, enable_graph_kernel=False): argument
|
| D | test_bfloat16.py | 41 def get_output(x0, x1, shape, enable_graph_kernel): argument
|
| D | test_bias_add_grad.py | 34 def get_output(dout, enable_graph_kernel=False): argument
|
| D | test_relu.py | 32 def get_output(x, enable_graph_kernel=False): argument
|
| D | test_assign_add.py | 35 def get_output(x2, y2, enable_graph_kernel=False): argument
|
| D | test_square_sum_all.py | 31 def get_output(inp0, inp1, enable_graph_kernel=False): argument
|
| D | test_slice.py | 33 def get_output(x, begin, size, 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_softmax.py | 30 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_dvm_slice.py | 40 def get_output(net, i0, i1, 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_equal_count.py | 33 def get_output(*inp, enable_graph_kernel=False): argument
|