| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | scale.c | 56 TensorC *scale_tensor = scale->base_.in_[SECOND_INPUT]; in ScaleInitInputDataType() local
120 TensorC *scale_tensor = scale->base_.in_[SECOND_INPUT]; in ScaleCalculateParameter() local
145 TensorC *scale_tensor = scale->base_.in_[SECOND_INPUT]; in ScaleInitScaleOffset() local
241 TensorC *scale_tensor = self->in_[SECOND_INPUT]; in ScaleResize() local
283 TensorC *scale_tensor = self->in_[SECOND_INPUT]; in ScaleCompute() local
|
| D | fused_batch_norm.c | 112 TensorC *scale_tensor = fused_batch_norm->bn_.base_.in_[SECOND_INPUT]; in FusedBatchNormBatchnorm2Scale() local
132 TensorC *scale_tensor = fused_batch_norm->bn_.base_.in_[SECOND_INPUT]; in FusedBatchNormInitConstTensor() local
208 TensorC *scale_tensor = fbn->bn_.base_.in_[SECOND_INPUT]; in FusedBatchNormTrainComputeInit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | scale_fp32_coder.cc | 27 Tensor *scale_tensor = input_tensors_.at(kWeightIndex); in InitScaleOffset() local
45 Tensor *scale_tensor = input_tensors_.at(kWeightIndex); in CalculateParameter() local
|
| D | groupnorm_fp32_coder.cc | 65 Tensor *scale_tensor = input_tensors_.at(kWeightIndex); in DoCode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | quant_batch_matmul.cc | 30 … const BaseTensorPtr &scale_tensor, const std::optional<BaseTensorPtr> &offset_tensor, in QuantMatmulV3AscendCall()
41 … const BaseTensorPtr &x2_tensor, const BaseTensorPtr &scale_tensor, in QuantMatmulV3AscendCustomize()
|
| D | quant_v2.cc | 31 const BaseTensorPtr &scale_tensor, const BaseTensorPtr &offset_tensor, in QuantV2AscendCustomize()
|
| D | ffn_ext.cc | 33 …const std::optional<BaseTensorPtr> &scale_tensor, const std::optional<BaseTensorPtr> &offset_tenso… in FFNExtAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/ |
| D | scale.cc | 52 auto scale_tensor = in_tensors_.at(INPUT_SCALE); in CheckSpecs() local
94 auto *scale_tensor = in_tensors_[INPUT_SCALE]; in InitWeights() local
169 auto scale_tensor = in_tensors_.at(INPUT_SCALE); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/ |
| D | resize_coreml.cc | 48 auto scale_tensor = in_tensors_.at(1); in BuildLayer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/op/ |
| D | instance_norm_nnapi.cc | 28 auto scale_tensor = in_tensors_.at(1); in InitParams() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/ |
| D | batchnorm_npu.cc | 55 auto scale_tensor = ConverterToNPUTensor(in_tensors[SCALE_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 | 63 auto scale_tensor = CreateInTensor("./bngrad/scale_3.bin", {1, 1, 1, channels}); in TEST_F() local
133 lite::Tensor scale_tensor(TypeId::kNumberTypeFloat32, {1, 1, 1, channels}); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | scale_tensorrt.cc | 165 auto scale_tensor = ConvertConstantTensorWithDims(ctx, in_tensors_[1], expect_shape, op_name_); in RunAsMutiDimsScale() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/pass/ |
| D | npu_pass_utils.cc | 180 auto scale_tensor = op->inputs().at(1); in Scale4dCase() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | batchnorm_fp32.c | 29 TensorC *scale_tensor = fused_batch_norm->bn_.base_.in_[SECOND_INPUT]; in FusedBatchNormEval() local
|