Home
last modified time | relevance | path

Searched defs:bias_tensor (Results 1 – 25 of 41) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Ddense.cc130 const auto &bias_tensor = input_values[kInputIndex2]->cast<tensor::BaseTensorPtr>(); in InferType() local
166 const auto &bias_tensor = input_values[kInputIndex2]->cast<tensor::BaseTensorPtr>(); in InferShape() local
185 const auto &bias_tensor = input_values[kInputIndex2]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dgroup_norm.cc110 const auto &bias_tensor = input_values[kInputIndex3]->cast<tensor::BaseTensorPtr>(); in InferType() local
135 const auto &bias_tensor = input_values[kInputIndex3]->cast<tensor::BaseTensorPtr>(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dquant_batch_matmul.cc31const std::optional<BaseTensorPtr> &bias_tensor, const bool &transpose_x1, in QuantMatmulV3AscendCall()
43 const std::optional<BaseTensorPtr> &bias_tensor, in QuantMatmulV3AscendCustomize()
Dweight_quant_batch_matmul.cc35const std::optional<BaseTensorPtr> &bias_tensor, int64_t antiquant_group_size, in WeightQuantBatchMatmulV2AscendCall()
64const std::optional<BaseTensorPtr> &bias_tensor, const BoolImmPtr &transpose_x, const BoolImmPtr &… in WeightQuantBatchMatmulV2AscendCustomize()
Dbatch_norm_ext.cc28 … const BaseTensorPtr &weight_tensor, const BaseTensorPtr &bias_tensor, in BatchNormExtAscendCustomize()
Dconvolution.cc28 const std::optional<BaseTensorPtr> &bias_tensor, in ConvolutionAscendCustomize()
Ddense.cc49 … const BaseTensorPtr &weight_tensor, const std::optional<BaseTensorPtr> &bias_tensor) { in DenseAscendCustomize()
Dconvolution_grad.cc29 …const BaseTensorPtr &weight_tensor, const std::optional<BaseTensorPtr> &bias_tensor, const ValueTu… in ConvolutionGradAscendCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/op/
Dinstance_norm_nnapi.cc29 auto bias_tensor = in_tensors_.at(2); in InitParams() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/
Dconvolution_base_coreml.cc46 auto bias_tensor = in_tensors_.at(kBiasIndex); in SetConvBias() local
Dmatmul_coreml.cc90 auto bias_tensor = in_tensors_.at(THIRD_INPUT); in ConstMatMulWithTransB() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/
Dconvolution_base_npu.cc131 std::shared_ptr<ge::Tensor> bias_tensor = ConverterToNPUTensor(inputs[BIAS_INDEX]); in InitBiasConst() local
Dfullconnection_int8_npu.cc100 auto bias_tensor = ConverterToNPUTensor(in_tensors[kBiasIndex], true); in SetNPUInputs() local
Dmatmul_npu.cc124 auto bias_tensor = ConverterToNPUTensor(in_tensors[BIAS_INDEX]); in SetNPUInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/pyboost/customize/
Ddense.cc47 … const BaseTensorPtr &weight_tensor, const std::optional<BaseTensorPtr> &bias_tensor) { in DenseGPUCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/
Ddense.cc48 … const BaseTensorPtr &weight_tensor, const std::optional<BaseTensorPtr> &bias_tensor) { in DenseCPUCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/bolt/
Dconvolution_bolt.cc103 auto bias_tensor = in_tensors_[kBiasIndex]; in InitWeightBiasTensor() local
417 auto bias_tensor = bias_tensors_[0]; in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Ddeconvolution_depthwise_int8.cc68 auto bias_tensor = in_tensors_.at(kBiasIndex); in InitWeightBias() local
Dconvolution_depthwise_int8.cc98 auto bias_tensor = in_tensors_.at(kBiasIndex); in InitWeightBias() local
Dconvolution_depthwise_3x3_int8.cc106 auto bias_tensor = in_tensors_.at(kBiasIndex); in InitWeightBias() local
Ddeconvolution_int8.cc168 auto bias_tensor = in_tensors_.at(kBiasIndex); in InitBiasWeight() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dbatchmatmul_fusion.cc177 auto bias_tensor = utils::cast<tensor::TensorPtr>(bias_default_param); in IsFCNonBias() local
186 auto bias_tensor = utils::cast<tensor::TensorPtr>(bias_value); in IsFCNonBias() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dbias_correction_strategy.cc463 …rrectionStrategy::AddBiasToInt32Tensor(const CNodePtr &cnode, const tensor::TensorPtr &bias_tensor, in AddBiasToInt32Tensor()
499 …orrectionStrategy::AddBiasToFp32Tensor(const CNodePtr &cnode, const tensor::TensorPtr &bias_tensor, in AddBiasToFp32Tensor()
528 auto bias_tensor = bias_default_param->cast<tensor::TensorPtr>(); in DoCNodeBiasCorrection() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_sparse/
Dmatmul_sparse_fp32.cc137 auto bias_tensor = in_tensors_[bias_tensor_index]; in PrepareBias() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Dmatmul_base_fp16.cc89 auto bias_tensor = in_tensors_[THIRD_INPUT]; in InitBias() local

12