Home
last modified time | relevance | path

Searched defs:base_operator (Results 1 – 25 of 253) sorted by relevance

1234567891011

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/ops/operator_populate/
Darithmetic_operator_populate.cc80 OpParameter *PopulateArithmeticCommonOpPara(const BaseOperatorPtr &base_operator) { in PopulateArithmeticCommonOpPara()
92 OpParameter *PopulateArithmeticFusionOpParameter(const BaseOperatorPtr &base_operator) { in PopulateArithmeticFusionOpParameter()
105 OpParameter *PopulateEltwiseOpParameter(const BaseOperatorPtr &base_operator) { in PopulateEltwiseOpParameter()
Dnllloss_populate.cc28 OpParameter *PopulateNLLLossOpParameter(const BaseOperatorPtr &base_operator) { in PopulateNLLLossOpParameter()
43 OpParameter *PopulateNLLLossGradOpParameter(const BaseOperatorPtr &base_operator) { in PopulateNLLLossGradOpParameter()
Dpooling_populate.cc76 OpParameter *PopulateAvgPoolOpParameter(const BaseOperatorPtr &base_operator) { in PopulateAvgPoolOpParameter()
130 OpParameter *PopulateMaxPoolOpParameter(const BaseOperatorPtr &base_operator) { in PopulateMaxPoolOpParameter()
Doperator_populate_register.h57 OpParameter *CreatePopulateByOp(const BaseOperatorPtr &base_operator) { in CreatePopulateByOp()
99 OpParameter *PopulateOpParameter(const BaseOperatorPtr &base_operator) { in PopulateOpParameter()
Darithmetic_self_populate.cc55 OpParameter *PopulateArithmeticSelfOpParameter(const BaseOperatorPtr &base_operator) { in PopulateArithmeticSelfOpParameter()
Dcall_populate.cc23 OpParameter *PopulateCallOpParameter(const BaseOperatorPtr &base_operator) { in PopulateCallOpParameter()
Dmatmul_populate.cc30 OpParameter *PopulateMatMulOpParameter(const BaseOperatorPtr &base_operator) { in PopulateMatMulOpParameter()
Dexp_populate.cc30 OpParameter *PopulateExpOpParameter(const BaseOperatorPtr &base_operator) { in PopulateExpOpParameter()
Dcommon_populate.cc33 OpParameter *PopulateCommonOpParameter(const BaseOperatorPtr &base_operator) { in PopulateCommonOpParameter()
Dattention_populate.cc23 OpParameter *PopulateAttentionOpParameter(const BaseOperatorPtr &base_operator) { in PopulateAttentionOpParameter()
Dcrop_and_resize_populate.cc23 OpParameter *PopulateCropAndResizeOpParameter(const BaseOperatorPtr &base_operator) { in PopulateCropAndResizeOpParameter()
Dsparse_softmax_cross_entropy_with_logits_populate.cc24 …ter *PopulateSparseSoftmaxCrossEntropyWithLogitsOpParameter(const BaseOperatorPtr &base_operator) { in PopulateSparseSoftmaxCrossEntropyWithLogitsOpParameter()
Dlocal_response_normalization_populate.cc23 OpParameter *PopulateLocalResponseNormOpParameter(const BaseOperatorPtr &base_operator) { in PopulateLocalResponseNormOpParameter()
Drange_populate.cc24 OpParameter *PopulateRangeOpParameter(const BaseOperatorPtr &base_operator) { in PopulateRangeOpParameter()
Dstrided_slice_operator_populate.cc23 OpParameter *PopulateStridedSliceOpParameter(const BaseOperatorPtr &base_operator) { in PopulateStridedSliceOpParameter()
Dfused_batchnorm_populate.cc23 OpParameter *PopulateFusedBatchNormOpParameter(const BaseOperatorPtr &base_operator) { in PopulateFusedBatchNormOpParameter()
Dtensor_list_reserve_populate.cc23 OpParameter *PopulateTensorListReserveOpParameter(const BaseOperatorPtr &base_operator) { in PopulateTensorListReserveOpParameter()
Dinstance_norm_populate.cc25 OpParameter *PopulateInstanceNormOpParameter(const BaseOperatorPtr &base_operator) { in PopulateInstanceNormOpParameter()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dslice_tensorrt.cc31 bool IsSupport(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &in_tensors, in IsSupport()
84 …vinfer1::Dims, nvinfer1::Dims, nvinfer1::Dims> GetSliceParams(const BaseOperatorPtr &base_operator, in GetSliceParams()
163 bool IsSupport(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &in_tensors, in IsSupport()
175 …vinfer1::Dims, nvinfer1::Dims, nvinfer1::Dims> GetSliceParams(const BaseOperatorPtr &base_operator, in GetSliceParams()
199 bool IsSupport(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &in_tensors, in IsSupport()
217 …vinfer1::Dims, nvinfer1::Dims, nvinfer1::Dims> GetSliceParams(const BaseOperatorPtr &base_operator, in GetSliceParams()
259 SliceTensorRT::SliceTensorRT(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &i… in SliceTensorRT()
279 int SliceTensorRT::IsSupport(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &i… in IsSupport()
Dinstancenorm_tensorrt.h25 …InstanceNormTensorRT(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &in_tenso… in InstanceNormTensorRT()
Dfill_tensorrt.h26 FillTensorRT(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &in_tensors, in FillTensorRT()
Dactivation_tensorrt.h25 …ActivationTensorRT(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &in_tensors, in ActivationTensorRT()
Dscatternd_tensorrt.h26 ScatterNdTensorRT(const BaseOperatorPtr &base_operator, const std::vector<TensorInfo> &in_tensors, in ScatterNdTensorRT()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mock/lite_runtime/populate/
Darithmetic_populate.cc41 ArithmeticParameter *PopulateArithmeticCommonPara(void *base_operator) { in PopulateArithmeticCommonPara()
65 OpParameter *PopulateArithmetic(void *base_operator) { in PopulateArithmetic()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/mapper/
Dop_mapper.cc31 STATUS SetOpInputs(const api::CNodePtr &cnode, mapper::BaseOperator *base_operator) { in SetOpInputs()
67 STATUS FillMultiOutOpOutputs(const api::CNodePtr &cnode, mapper::BaseOperator *base_operator, in FillMultiOutOpOutputs()
116 STATUS SetOpOutputs(const api::CNodePtr &cnode, mapper::BaseOperator *base_operator, in SetOpOutputs()
146 STATUS SetCommonAttr(const api::CNodePtr &cnode, mapper::BaseOperator *base_operator, in SetCommonAttr()
164 STATUS SetConvFcDataInfo(const api::CNodePtr &cnode, mapper::BaseOperator *base_operator) { in SetConvFcDataInfo()
338 STATUS PushOfflineArgs(const api::CNodePtr &cnode, mapper::BaseOperator *base_operator, size_t offl… in PushOfflineArgs()

1234567891011