| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | img2col_impl.py | 37 def shape56_0(tik_instance, input_x, res, input_shape, shape_info): argument
64 def shape56_1(tik_instance, input_x, res, input_shape, shape_info): argument
92 def shape56_2(tik_instance, input_x, res, input_shape, shape_info): argument
116 def shape56_3(tik_instance, input_x, res, input_shape, shape_info): argument
144 def shape56_4(tik_instance, input_x, res, input_shape, shape_info): argument
173 def shape28_0(tik_instance, input_x, res, input_shape, shape_info): argument
201 def shape28_1(tik_instance, input_x, res, input_shape, shape_info): argument
240 def shape28_2(tik_instance, input_x, res, input_shape, shape_info): argument
288 def shape28_3(tik_instance, input_x, res, input_shape, shape_info): argument
316 def shape28_4(tik_instance, input_x, res, input_shape, shape_info): argument
[all …]
|
| D | transpose02314_impl.py | 61 def cus_transpose02314(input_x, output, kernel_name="cus_transpose021354"): argument
104 def shape0(tik_instance, input_x, res, dtype): argument
124 def shape1(tik_instance, input_x, res, dtype): argument
153 def shape2(tik_instance, input_x, res, dtype): argument
173 def shape3(tik_instance, input_x, res, dtype): argument
193 def shape4(tik_instance, input_x, res, dtype): argument
213 def shape5(tik_instance, input_x, res, dtype): argument
232 def shape6(tik_instance, input_x, res, dtype): argument
261 def shape7(tik_instance, input_x, res, dtype): argument
277 def shape8(tik_instance, input_x, res, dtype): argument
[all …]
|
| D | fused_abs_max1_impl.py | 76 def shape0(tik_instance, input_x_shape, input_x, res): argument
100 def shape1(tik_instance, input_x_shape, ori_shape, input_x, res): argument
146 def shape2(tik_instance, input_x_shape, input_x, res): argument
171 def shape3_1000(tik_instance, input_x, res): argument
207 def shape3_1001(tik_instance, input_x, res): argument
242 def shape3(tik_instance, input_x_shape, ori_shape, input_x, res): argument
275 def shape4(tik_instance, input_x_shape, input_x, res): argument
302 def shape5(tik_instance, input_x_shape, input_x, res): argument
330 def shape6(tik_instance, ori_shape, input_x, res): argument
362 def shape7(tik_instance, input_x_shape, input_x, res): argument
[all …]
|
| D | fake_learned_scale_quant_perlayer.py | 74 def fake_learned_scale_quant_perlayer_param(input_x, alpha, quant_max, argument
111 def fake_learned_scale_quant_perlayer(input_x, alpha, quant_max, out, neg_trunc, argument
|
| D | fake_learned_scale_quant_perchannel.py | 74 def fake_learned_scale_quant_perchannel_param(input_x, alpha, quant_max, channel_axis, argument
119 def fake_learned_scale_quant_perchannel(input_x, alpha, quant_max, out, neg_trunc, channel_axis, argument
|
| D | fake_learned_scale_quant_perlayer_grad.py | 144 def fake_learned_scale_quant_perlayer_grad_d_param(input_x, alpha, quant_max, argument
182 def fake_learned_scale_quant_perlayer_grad_d(dout, input_x, alpha, quant_max, dx, dalpha, neg_trunc, argument
|
| D | fake_learned_scale_quant_perchannel_grad.py | 143 def fake_learned_scale_quant_perchannel_grad_d_param(input_x, alpha, quant_max, channel_axis, argument
189 def fake_learned_scale_quant_perchannel_grad_d(dout, input_x, alpha, quant_max, dx, dalpha, neg_tru… argument
|
| D | matrix_combine_impl.py | 34 def cus_matrix_combine(input_x, output, kernel_name="cus_matrix_combine"): argument
|
| D | cholesky_trsm_impl.py | 34 def cus_cholesky_trsm(input_x, output, kernel_name): argument
|
| /third_party/mindspore/tests/ut/python/pipeline/parse/ |
| D | test_call_innetr_net_attr.py | 33 def construct(self, input_x): argument
46 def construct(self, input_x): argument
61 def construct(self, input_x, input_y): argument
72 def construct(self, input_x, input_y): argument
83 def construct(self, input_x, input_y): argument
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_effect_ops.py | 81 def __init__(self, input_x): argument
106 def __init__(self, input_x): argument
131 def __init__(self, input_x): argument
156 def __init__(self, input_x): argument
181 def __init__(self, input_x): argument
206 def __init__(self, input_x): argument
231 def __init__(self, input_x): argument
256 def __init__(self, input_x): argument
281 def __init__(self, input_x): argument
306 def __init__(self, input_x): argument
[all …]
|
| /third_party/mindspore/tests/st/ops/custom_ops_tbe/ |
| D | square_impl.py | 26 def square_compute(input_x, output_y): argument
64 def CusSquareImpl(input_x, output_y, kernel_name="CusSquareImpl"): argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_logical_op.py | 30 def construct(self, input_x, input_y): argument
39 def construct(self, input_x, input_y): argument
48 def construct(self, input_x): argument
|
| /third_party/mindspore/mindspore/nn/probability/distribution/_utils/ |
| D | custom_ops.py | 21 def exp_generic(input_x): argument
36 def log_generic(input_x): argument
|
| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_train.py | 74 def construct(self, input_x): argument
106 def construct(self, input_x): argument
136 def construct(self, input_x): argument
169 def construct(self, input_x): argument
|
| D | test_pooling.py | 31 def construct(self, input_x): argument
43 def construct(self, input_x): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | select_impl.cu | 23 __global__ void Select(const size_t size, const bool* cond, const T* input_x, const T* input_y, T* … in Select()
31 void CalSelect(const size_t size, const bool* cond, const T* input_x, const T* input_y, T* output, in CalSelect()
|
| D | loss_with_reduction_impl.cu | 137 __global__ void KLDivLossKernel(const int input_size, const int reduction, const T *input_x, const … in KLDivLossKernel()
156 void KLDivLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y, T *… in KLDivLoss()
176 __global__ void KLDivLossGradKernel(const int input_size, const int reduction, const T *input_x, co… in KLDivLossGradKernel()
200 void KLDivLossGrad(const int &input_size, const int &reduction, const T *input_x, const T *input_y,… in KLDivLossGrad()
207 …al__ void BinaryCrossEntropyLossKernel(const int input_size, const int reduction, const T *input_x, in BinaryCrossEntropyLossKernel()
237 void BinaryCrossEntropyLoss(const int &input_size, const int &reduction, const T *input_x, const T … in BinaryCrossEntropyLoss()
257 … void BinaryCrossEntropyLossGradKernel(const int input_size, const int reduction, const T *input_x, in BinaryCrossEntropyLossGradKernel()
297 void BinaryCrossEntropyLossGrad(const int &input_size, const int &reduction, const T *input_x, cons… in BinaryCrossEntropyLossGrad()
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_conv_bn1_fusion.py | 71 def construct(self, input_x): argument
99 def construct(self, input_x): argument
125 def construct(self, input_x): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | binary_cross_entropy.c | 20 … void BinaryCrossEntropyLossKernel(const int input_size, const int reduction, const float *input_x, in BinaryCrossEntropyLossKernel()
38 void BinaryCrossEntropy(const int input_size, const int reduction, const float *input_x, const floa… in BinaryCrossEntropy()
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_math_ops.py | 42 def bprop(input_x, input_y, out, dout): argument
92 def bprop(input_x, out, dout): argument
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_batchnorm_grad.py | 28 …def construct(self, input_dy, input_x, input_scale, input_save_mean, input_save_inv_variance, inpu… argument
33 def get_output(input_dy, input_x, input_scale, input_save_mean, input_save_inv_variance, input_reve… argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | minimum_cpu_kernel.cc | 84 void MinimumCPUKernel<T>::BroadcastArith(const T *input_x, const T *input_y, T *output) const { in BroadcastArith() argument
157 … const size_t d6, const T *input_x, const T *input_y, T *output) const { in BroadcastArithKernel() argument
186 void MinimumCPUKernel<T>::BroadcastArithOneScalarOneTensor(const T *input_x, const T *input_y, T *o… in BroadcastArithOneScalarOneTensor() argument
199 void MinimumCPUKernel<T>::BroadcastArithTensors(const T *input_x, const T *input_y, T *output) cons… in BroadcastArithTensors() argument
|
| D | maximum_cpu_kernel.cc | 84 void MaximumCPUKernel<T>::BroadcastArith(const T *input_x, const T *input_y, T *output) const { in BroadcastArith() argument
157 … const size_t d6, const T *input_x, const T *input_y, T *output) const { in BroadcastArithKernel() argument
186 void MaximumCPUKernel<T>::BroadcastArithOneScalarOneTensor(const T *input_x, const T *input_y, T *o… in BroadcastArithOneScalarOneTensor() argument
199 void MaximumCPUKernel<T>::BroadcastArithTensors(const T *input_x, const T *input_y, T *output) cons… in BroadcastArithTensors() argument
|
| /third_party/mindspore/tests/ut/python/train/ |
| D | test_amp.py | 41 def construct(self, input_x, label): argument
52 def construct(self, input_x): argument
|