Searched defs:dout_tensor (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/ |
| D | silu_grad.cc | 30 OpPtr SiLUGradCPUCall(const device::DeviceContext *device_context, const BaseTensorPtr &dout_tensor, in SiLUGradCPUCall()
51 void SiLUGradCPUCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &dout_tensor, in SiLUGradCPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | smooth_l1_loss_grad_v2.cc | 78 Tensor *dout_tensor = ctx.Input(2); in ParamCheck() local
132 Tensor *dout_tensor = ctx.Input(2); in AttributesCheck() local
192 Tensor *dout_tensor = ctx.Input(2); in ComputeSum() local
260 Tensor *dout_tensor = ctx.Input(2); in ComputeMean() local
332 Tensor *dout_tensor = ctx.Input(2); in ComputeNone() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | batch_norm_grad_ext.cc | 27 …chNormGradExtAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &dout_tensor, in BatchNormGradExtAscendCustomize()
|
| D | group_norm_grad.cc | 32 …GroupNormGradAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &dout_tensor, in GroupNormGradAscendCustomize()
|
| D | convolution_grad.cc | 28 …const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &dout_tensor, const BaseTensorPtr &input_… in ConvolutionGradAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | op_utils.cc | 203 const auto &dout_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in EltwiseGradSimpleInferShape() local
227 const auto &dout_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in EltwiseGradSimpleInferType() local
|