Searched defs:offset_tensor (Results 1 – 11 of 11) sorted by relevance
30 … const BaseTensorPtr &scale_tensor, const std::optional<BaseTensorPtr> &offset_tensor, in QuantMatmulV3AscendCall()42 … const std::optional<BaseTensorPtr> &offset_tensor, in QuantMatmulV3AscendCustomize()
31 const BaseTensorPtr &scale_tensor, const BaseTensorPtr &offset_tensor, in QuantV2AscendCustomize()
33 …const std::optional<BaseTensorPtr> &scale_tensor, const std::optional<BaseTensorPtr> &offset_tenso… in FFNExtAscendCustomize()
71 TensorC *offset_tensor = scale->base_.in_[SECOND_INPUT]; in ScaleInitInputDataType() local180 TensorC *offset_tensor = scale->base_.in_[THIRD_INPUT]; in ScaleInitScaleOffset() local290 TensorC *offset_tensor = self->in_[THIRD_INPUT]; in ScaleCompute() local
115 TensorC *offset_tensor = fused_batch_norm->bn_.base_.in_[THIRD_INPUT]; in FusedBatchNormBatchnorm2Scale() local133 TensorC *offset_tensor = fused_batch_norm->bn_.base_.in_[THIRD_INPUT]; in FusedBatchNormInitConstTensor() local209 TensorC *offset_tensor = fbn->bn_.base_.in_[THIRD_INPUT]; in FusedBatchNormTrainComputeInit() local
66 Tensor *offset_tensor = input_tensors_.at(kBiasIndex); in DoCode() local
64 auto offset_tensor = ConverterToNPUTensor(in_tensors[OFFSET_INDEX]); in SetNPUInputs() local
70 auto offset_tensor = input_tensors_.at(kBiasIndex); in DoCode() local
73 auto offset_tensor = in_tensors_.at(kOffsetIndex); in InitScaleOffset() local
95 auto *offset_tensor = in_tensors_[INPUT_OFFSET]; in InitWeights() local
30 TensorC *offset_tensor = fused_batch_norm->bn_.base_.in_[THIRD_INPUT]; in FusedBatchNormEval() local