| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_low_precision.py | 42 def get_output(x, y, z, net, enable_graph_kernel=False): function
73 def get_output(x, y, net, enable_graph_kernel=False): function
|
| D | test_recompute.py | 96 def get_output(i0, net, enable_graph_kernel=False): function
114 def get_output(i0, i1, net, enable_graph_kernel=False): function
132 def get_output(i0, i1, i2, net, enable_graph_kernel=False): function
|
| D | test_addn.py | 33 def get_output(*tensors): function
|
| D | test_identity.py | 31 def get_output(x, enable_graph_kernel=False): function
|
| D | test_equal_count.py | 32 def get_output(*inp, enable_graph_kernel=False): function
|
| D | test_argmax.py | 33 def get_output(x, axis, enable_graph_kernel=False): function
|
| D | test_bias_add_grad.py | 33 def get_output(dout, enable_graph_kernel=False): function
|
| D | test_square_sum_all.py | 31 def get_output(inp0, inp1, enable_graph_kernel=False): function
|
| D | test_assign_add.py | 35 def get_output(x2, y2, enable_graph_kernel=False): function
|
| D | test_relu.py | 32 def get_output(x, enable_graph_kernel=False): function
|
| D | test_reduce_min.py | 33 def get_output(x, axis, keep_dims, enable_graph_kernel=False): function
|
| D | test_softmax.py | 30 def get_output(x, enable_graph_kernel=False): function
|
| D | test_cast_matmul_fusion.py | 37 def get_output(i0, i1, i2, enable_graph_kernel=False): function
|
| D | test_slice.py | 33 def get_output(x, begin, size, enable_graph_kernel=False): function
|
| D | test_reduce_max.py | 33 def get_output(x, axis, keep_dims, enable_graph_kernel=False): function
|
| D | test_squeeze.py | 32 def get_output(x, axis=(), enable_graph_kernel=False): function
|
| D | test_squared_difference.py | 32 def get_output(x, y, enable_graph_kernel=False): function
|
| D | test_fuse.py | 43 def get_output(i0, i1, i2, enable_graph_kernel=False): function
|
| D | test_tile.py | 33 def get_output(x, multiples, enable_graph_kernel=False): function
|
| D | test_expand_dims.py | 32 def get_output(x, axis, enable_graph_kernel=False): function
|
| D | test_sqrt_grad.py | 32 def get_output(x, dout, enable_graph_kernel=False): function
|
| D | test_relu_grad.py | 32 def get_output(y_backprop, x, enable_graph_kernel=False): function
|
| D | test_bias_add.py | 32 def get_output(x, b, data_format, enable_graph_kernel): function
|
| D | test_lamb_apply_weight_assign.py | 35 def get_output(w_norm, g_norm, lr, update, param, enable_graph_kernel=False): function
|
| D | test_oneslike.py | 32 def get_output(shape, dtype, nptype, enable_graph_kernel=False): function
|