| /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 …vLossKernel(const int input_size, const int reduction, const T *input_x, const T *input_y, T *loss, in KLDivLossKernel()
156 void KLDivLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y, T *… in KLDivLoss()
176 … KLDivLossGradKernel(const int input_size, const int reduction, const T *input_x, const T *input_y, in KLDivLossGradKernel()
200 void KLDivLossGrad(const int &input_size, const int &reduction, const T *input_x, const T *input_y,… in KLDivLossGrad()
208 … const T *input_y, const T *weight, T *loss, T *tmp_loss) { in BinaryCrossEntropyLossKernel()
237 …ryCrossEntropyLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y, in BinaryCrossEntropyLoss()
258 … const T *input_y, const T *weight, const T *dloss, T *dx) { in BinaryCrossEntropyLossGradKernel()
297 …ossEntropyLossGrad(const int &input_size, const int &reduction, const T *input_x, const T *input_y, in BinaryCrossEntropyLossGrad()
|
| /third_party/mindspore/tests/ut/python/pipeline/parse/ |
| D | test_call_innetr_net_attr.py | 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/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | binary_cross_entropy.c | 21 … const float *input_y, const float *weight, float *loss, float *tmp_loss) { in BinaryCrossEntropyLossKernel()
38 …CrossEntropy(const int input_size, const int reduction, const float *input_x, const float *input_y, in BinaryCrossEntropy()
|
| D | binary_cross_entropy_grad.c | 21 …sEntropyGrad(const int input_size, const int reduction, const float *input_x, const float *input_y, in BinaryCrossEntropyGrad()
|
| /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
|
| D | test_select_op.py | 29 def construct(self, cond_op, input_x, input_y): 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
|
| D | select_cpu_kernel.cc | 44 auto *input_y = reinterpret_cast<T *>(inputs[2]->addr); in Launch() local
|
| D | dynamic_assign_cpu_kernel.cc | 81 auto *input_y = reinterpret_cast<T *>(inputs[1]->addr); in LaunchKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/expanders/ |
| D | bias_add.cc | 44 auto input_y = inputs[1]; in Expand() local
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_select_op.py | 30 def construct(self, cond_op, input_x, input_y): argument
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_maximum_grad.py | 44 def get_maximum_grad_output(input_x, input_y, input_dout, enable_graph_kernel=False): argument
|
| D | test_minimum_grad.py | 44 def get_minimum_grad_output(input_x, input_y, input_dout, enable_graph_kernel=False): argument
|
| D | test_gelu.py | 69 def gelu_grad(input_dy, input_x, input_y, enable_graph_kernel=False): argument
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_merge.cc | 39 …auto input_y = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2, … in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | kl_div_loss_grad_kernel.h | 44 T *input_y = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| D | kl_div_loss_gpu_kernel.h | 44 T *input_y = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| D | binary_cross_entropy_grad_kernel.h | 44 T *input_y = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| D | binary_cross_entropy_gpu_kernel.h | 42 T *input_y = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| /third_party/mindspore/tests/st/pynative/ms_function/ |
| D | test_nested_calling_ms_function.py | 28 input_y = Tensor(np.full((1, 1, 120, 640), 4), dtype=mstype.float32) variable
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_math_ops.py | 42 def bprop(input_x, input_y, out, dout): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | select_gpu_kernel.h | 43 T *input_y = GetDeviceAddress<T>(inputs, 2); in Launch() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/ |
| D | tensor_array.cc | 86 lite::Tensor *input_y = in_tensors_.at(kIndexInputIdx); in Init() local
|