Home
last modified time | relevance | path

Searched defs:input_args (Results 1 – 25 of 676) sorted by relevance

12345678910>>...28

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dmap_tensor_get_grad.cc36 const std::vector<AbstractBasePtr> &input_args) { in MapTensorGetGradInferInner()
52 …MapTensorGetGradInferShape(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in MapTensorGetGradInferShape()
56 … MapTensorGetGradInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in MapTensorGetGradInferType()
61 const std::vector<AbstractBasePtr> &input_args) { in MapTensorGetGradInfer()
DopaquePredicate.cc42 const std::vector<AbstractBasePtr> &input_args) { in OpaquePredicateInferShape()
48 …r OpaquePredicateInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in OpaquePredicateInferType()
64 const std::vector<AbstractBasePtr> &input_args) { in OpaquePredicateInfer()
Dincre_flash_attention.cc107 bool CheckIsFrontend(const std::vector<AbstractBasePtr> &input_args) { in CheckIsFrontend()
114 std::vector<int64_t> ObtainCorrShape(const std::vector<AbstractBasePtr> &input_args, size_t index) { in ObtainCorrShape()
125 …> GetIFADynInputShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in GetIFADynInputShape()
182 …PaddingAttenMaskShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in CheckPaddingAttenMaskShape()
214 …lSeqLengthsShapeValue(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in CheckActualSeqLengthsShapeValue()
237 … const std::vector<AbstractBasePtr> &input_args) { in IncreFlashAttentionInferShapeBSH()
278 … const std::vector<AbstractBasePtr> &input_args) { in IncreFlashAttentionInferShapeBNSD()
328 const std::vector<AbstractBasePtr> &input_args) { in IncreFlashAttentionInferShape()
337 void CheckQuantParamType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in CheckQuantParamType()
379 …creFlashAttentionInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in IncreFlashAttentionInferType()
[all …]
Dmap_tensor_put.cc40 …apTensorPutInferInner(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in MapTensorPutInferInner()
96 …Ptr MapTensorPutInferShape(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in MapTensorPutInferShape()
102 …ePtr MapTensorPutInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in MapTensorPutInferType()
108 const std::vector<AbstractBasePtr> &input_args) { in MapTensorPutInfer()
Dmap_tensor_get_data.cc39 const std::vector<AbstractBasePtr> &input_args) { in MapTensorGetDataInferInner()
68 const std::vector<AbstractBasePtr> &input_args) { in MapTensorGetDataInferShape()
75 … MapTensorGetDataInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in MapTensorGetDataInferType()
83 const std::vector<AbstractBasePtr> &input_args) { in MapTensorGetDataInfer()
Dselect_view.cc46 … SelectViewInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in SelectViewInferShape()
50 …r SelectViewInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in SelectViewInferType()
55 const std::vector<AbstractBasePtr> &input_args) { in SelectViewInfer()
Dmap_tensor_put_with_status.cc41 const std::vector<AbstractBasePtr> &input_args) { in MapTensorPutWithStatusInferInner()
118 const std::vector<AbstractBasePtr> &input_args) { in MapTensorPutWithStatusInferShape()
123 …nsorPutWithStatusInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in MapTensorPutWithStatusInferType()
129 const std::vector<AbstractBasePtr> &input_args) { in MapTensorPutWithStatusInfer()
Dunsorted_segment_arithmetic.cc52 …t GetNumSegmentsValue(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in GetNumSegmentsValue()
97 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
149 … const std::vector<AbstractBasePtr> &input_args) { in UnsortedSegmentArithmeticInferShape()
182 const std::vector<AbstractBasePtr> &input_args) { in UnsortedSegmentArithmeticInferType()
222 const std::vector<AbstractBasePtr> &input_args) { in UnsortedSegmentArithmeticInfer()
Dtrunc.cc42 …ract::ShapePtr TruncInferShape(const PrimitivePtr &, const std::vector<AbstractBasePtr> &input_arg… in TruncInferShape()
49 TypePtr TruncInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in TruncInferType()
61 const std::vector<AbstractBasePtr> &input_args) { in TruncInfer()
Dscatter_elements.cc37 …ePtr ScatterElementsInferShape(const PrimitivePtr &, const std::vector<AbstractBasePtr> &input_arg… in ScatterElementsInferShape()
41 …tterElementsInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in ScatterElementsInferType()
49 const std::vector<AbstractBasePtr> &input_args) { in ScatterElementsInfer()
Dhypot.cc31 …pePtr HypotInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in HypotInferShape()
35 TypePtr HypotInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in HypotInferType()
44 const std::vector<AbstractBasePtr> &input_args) { in HypotInfer()
Dtensor_copy.cc30 … TensorMoveInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in TensorMoveInferShape()
36 …r TensorMoveInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in TensorMoveInferType()
48 const std::vector<AbstractBasePtr> &input_args) { in TensorMoveInfer()
Dlayer_norm_x_backprop_v2.cc43 … const std::vector<AbstractBasePtr> &input_args) { in LayerNormXBackpropV2InferShape()
50 …erNormXBackpropV2InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in LayerNormXBackpropV2InferType()
63 const std::vector<AbstractBasePtr> &input_args) { in LayerNormXBackpropV2Infer()
Drelu_v3.cc32 …:BaseShapePtr ReLUV3InferShape(const PrimitivePtr &, const std::vector<AbstractBasePtr> &input_arg… in ReLUV3InferShape()
37 TypePtr ReLUV3InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in ReLUV3InferType()
47 const std::vector<AbstractBasePtr> &input_args) { in ReLUV3Infer()
Dsequence_addn.cc37 …equenceAddNInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in SequenceAddNInferShape()
43 …SequenceAddNInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in SequenceAddNInferType()
55 …equenceAddNInferInner(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in SequenceAddNInferInner()
Dlcm.cc40 …hapePtr LcmInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in LcmInferShape()
44 TypePtr LcmInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in LcmInferType()
53 const std::vector<AbstractBasePtr> &input_args) { in LcmInfer()
Dsparse_fill_empty_rows.cc51 bool CheckSparseFillEmptyRowsInputs(const std::vector<AbstractBasePtr> &input_args, const std::stri… in CheckSparseFillEmptyRowsInputs()
91 … const std::vector<AbstractBasePtr> &input_args) { in SparseFillEmptyRowsInferShape()
125 … const std::vector<AbstractBasePtr> &input_args) { in SparseFillEmptyRowsFrontendInferShape()
158 …illEmptyRowsInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in SparseFillEmptyRowsInferType()
181 const std::vector<AbstractBasePtr> &input_args) { in SparseFillEmptyRowsInfer()
Dbuffer_append.cc34 … const std::vector<AbstractBasePtr> &input_args) { in GetSequnceIndexShape()
60 const std::vector<AbstractBasePtr> &input_args) { in GetSequnceIndexType()
86 const std::vector<AbstractBasePtr> &input_args) { in BufferAppendInferShape()
130 …BufferAppendInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in BufferAppendInferType()
160 const std::vector<AbstractBasePtr> &input_args) { in BufferAppendInfer()
Dfused_sparse_proximal_adagrad.cc59 … const std::vector<AbstractBasePtr> &input_args, in FusedSparseProximalAdagradInferShapeCommon()
94 … const std::vector<AbstractBasePtr> &input_args) { in FusedSparseProximalAdagradInferShapeIner()
102 … const std::vector<AbstractBasePtr> &input_args) { in FusedSparseProximalAdagradInferShape()
109 …seProximalAdagradInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in FusedSparseProximalAdagradInferType()
145 const std::vector<AbstractBasePtr> &input_args) { in FusedSparseProximalAdagradInfer()
Dsparse_sparse_arithmetic.cc48 void CheckSparseSparseArithmeticInputs(const std::vector<AbstractBasePtr> &input_args, const std::s… in CheckSparseSparseArithmeticInputs()
119 … const std::vector<AbstractBasePtr> &input_args) { in SparseSparseArithmeticInferShape()
143 … const std::vector<AbstractBasePtr> &input_args) { in SparseSparseArithmeticFrontendInferShape()
170 const std::vector<AbstractBasePtr> &input_args) { in SparseSparseArithmeticInferType()
187 const std::vector<AbstractBasePtr> &input_args) { in SparseSparseArithmeticInfer()
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ops/
Dop_infer.h56 …l ValuePtr InferValue(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferValue()
73 const std::vector<AbstractBasePtr> &input_args) const { in InferShapeAndType()
86 const std::vector<AbstractBasePtr> &input_args) const { in InferAbstract()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dop_func_impl.h39 …seShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
59 …ual TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferType()
80 …t32_t CheckValidation(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in CheckValidation()
Dfft_arithmetic.cc27 …hapePtr FFTInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in FFTInferShape()
57 …apePtr FFTNInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in FFTNInferShape()
98 TypePtr FFTInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args)… in FFTInferType()
131 …_t FFTCheckValidation(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in FFTCheckValidation()
168 …t FFTNCheckValidation(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in FFTNCheckValidation()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/
Dscale_grad_fusion.cc27 const std::vector<AbstractBasePtr> &input_args) { in ScaleGradInferShape()
32 …tr ScaleGradInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in ScaleGradInferType()
39 const std::vector<AbstractBasePtr> &input_args) { in ScaleGradInferShapeAndType()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dmax_pool_grad.cc41 const std::vector<AbstractBasePtr> &input_args) { in MaxPoolGradInferShape()
47 … MaxPoolGradInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in MaxPoolGradInferType()
58 … const std::vector<abstract::AbstractBasePtr> &input_args) { in MaxPoolGradInfer()

12345678910>>...28