Searched defs:mean_tensor (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | normal_tensor_tensor.cc | 28 const BaseTensorPtr &mean_tensor, in NormalTensorTensorAscendCustomize()
|
| D | normal_tensor_float.cc | 28 … const BaseTensorPtr &mean_tensor, const FP32ImmPtr &std_float, in NormalTensorFloatAscendCustomize()
|
| D | layer_norm_grad_ext.cc | 29 … const BaseTensorPtr &mean_tensor, const BaseTensorPtr &variance_tensor, in LayerNormGradExtAscendCustomize()
|
| D | batch_norm_ext.cc | 29 … const BaseTensorPtr &mean_tensor, const BaseTensorPtr &variance_tensor, in BatchNormExtAscendCustomize()
|
| D | group_norm_grad.cc | 33 … const BaseTensorPtr &input_tensor, const BaseTensorPtr &mean_tensor, in GroupNormGradAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | batchnorm_fp32_coder.cc | 55 Tensor *mean_tensor = input_tensors_.at(1); in DoCode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/ |
| D | batchnorm_npu.cc | 73 auto mean_tensor = ConverterToNPUTensor(in_tensors[MEAN_INDEX]); in SetNPUInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | bn_grad_fp32_test.cc | 65 auto mean_tensor = CreateInTensor("./bngrad/save_mean_3.bin", {1, 1, 1, channels}); in TEST_F() local
143 lite::Tensor mean_tensor(TypeId::kNumberTypeFloat32, {1, 1, 1, channels}); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | batch_norm.c | 31 TensorC *mean_tensor = batch_norm->base_.in_[SECOND_INPUT]; in BatchNormFillParam() local
|
| D | fused_batch_norm.c | 134 TensorC *mean_tensor = fused_batch_norm->bn_.base_.in_[FOURTH_INPUT]; in FusedBatchNormInitConstTensor() local
210 TensorC *mean_tensor = fbn->bn_.base_.in_[FOURTH_INPUT]; in FusedBatchNormTrainComputeInit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | poisson.cc | 135 auto mean_tensor = ctx.Input(kPoissonMeanIndex); in ParseKernelParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | batchnorm_fp32.c | 31 TensorC *mean_tensor = fused_batch_norm->bn_.base_.in_[FOURTH_INPUT]; in FusedBatchNormEval() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | batchnorm_to_scale_fusion.cc | 283 auto mean_tensor = bn_mean_param->default_param()->cast<tensor::TensorPtr>(); in Run() local
|