| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir_1.cc | 27 auto ops_offset = schema::CreateDepend(fbb); in MindIR_Depend_CreatePrimitive() local
38 auto ops_offset = schema::CreateDropout(fbb, keep_prob); in MindIR_Dropout_CreatePrimitive() local
66 auto ops_offset = schema::CreateDropout(fbb, keep_prob); in MindIR_Dropout_SetKeepProb() local
79 auto ops_offset = schema::CreateElu(fbb, alpha); in MindIR_Elu_CreatePrimitive() local
107 auto ops_offset = schema::CreateElu(fbb, alpha); in MindIR_Elu_SetAlpha() local
120 auto ops_offset = schema::CreateEmbeddingLookupFusion(fbb, max_norm); in MindIR_EmbeddingLookupFusion_CreatePrimitive() local
148 auto ops_offset = schema::CreateEmbeddingLookupFusion(fbb, max_norm); in MindIR_EmbeddingLookupFusion_SetMaxNorm() local
161 auto ops_offset = schema::CreateFakeQuantWithMinMaxVars(fbb, num_bits, narrow_range); in MindIR_FakeQuantWithMinMaxVars_CreatePrimitive() local
189 auto ops_offset = schema::CreateFakeQuantWithMinMaxVars(fbb, num_bits, value->narrow_range()); in MindIR_FakeQuantWithMinMaxVars_SetNumBits() local
220 auto ops_offset = schema::CreateFakeQuantWithMinMaxVars(fbb, value->num_bits(), narrow_range); in MindIR_FakeQuantWithMinMaxVars_SetNarrowRange() local
[all …]
|
| D | mindir.cc | 28 …auto ops_offset = schema::CreateActivation(fbb, static_cast<schema::ActivationType>(activation_typ… in MindIR_Activation_CreatePrimitive() local
63 auto ops_offset = in MindIR_Activation_SetActivationType() local
99 …auto ops_offset = schema::CreateActivation(fbb, static_cast<schema::ActivationType>(value->activat… in MindIR_Activation_SetAlpha() local
134 …auto ops_offset = schema::CreateActivation(fbb, static_cast<schema::ActivationType>(value->activat… in MindIR_Activation_SetMinVal() local
169 …auto ops_offset = schema::CreateActivation(fbb, static_cast<schema::ActivationType>(value->activat… in MindIR_Activation_SetMaxVal() local
204 …auto ops_offset = schema::CreateActivation(fbb, static_cast<schema::ActivationType>(value->activat… in MindIR_Activation_SetApproximate() local
220 …auto ops_offset = schema::CreateAddFusion(fbb, static_cast<schema::ActivationType>(activation_type… in MindIR_AddFusion_CreatePrimitive() local
250 …auto ops_offset = schema::CreateAddFusion(fbb, static_cast<schema::ActivationType>(activation_type… in MindIR_AddFusion_SetActivationType() local
264 auto ops_offset = schema::CreateArgMaxFusion(fbb, axis, top_k, keep_dims, out_max_value); in MindIR_ArgMaxFusion_CreatePrimitive() local
292 auto ops_offset = in MindIR_ArgMaxFusion_SetAxis() local
[all …]
|
| D | mindir_2.cc | 27 …auto ops_offset = schema::CreateSplitWithOverlap(fbb, split_dim, number_split, fbb.CreateVector(ra… in MindIR_SplitWithOverlap_CreatePrimitive() local
55 …auto ops_offset = schema::CreateSplitWithOverlap(fbb, split_dim, value->number_split(), fbb.Create… in MindIR_SplitWithOverlap_SetSplitDim() local
86 …auto ops_offset = schema::CreateSplitWithOverlap(fbb, value->split_dim(), number_split, fbb.Create… in MindIR_SplitWithOverlap_SetNumberSplit() local
124 …auto ops_offset = schema::CreateSplitWithOverlap(fbb, value->split_dim(), value->number_split(), f… in MindIR_SplitWithOverlap_SetRatio() local
162 …auto ops_offset = schema::CreateSplitWithOverlap(fbb, value->split_dim(), value->number_split(), f… in MindIR_SplitWithOverlap_SetExtendTop() local
200 …auto ops_offset = schema::CreateSplitWithOverlap(fbb, value->split_dim(), value->number_split(), f… in MindIR_SplitWithOverlap_SetExtendBottom() local
214 …auto ops_offset = schema::CreateGenOP(fbb, static_cast<schema::ActivationType>(activation_type), a… in MindIR_GenOP_CreatePrimitive() local
244 …auto ops_offset = schema::CreateGenOP(fbb, static_cast<schema::ActivationType>(activation_type), v… in MindIR_GenOP_SetActivationType() local
275 …auto ops_offset = schema::CreateGenOP(fbb, static_cast<schema::ActivationType>(value->activation_t… in MindIR_GenOP_SetAlpha() local
306 …auto ops_offset = schema::CreateGenOP(fbb, static_cast<schema::ActivationType>(value->activation_t… in MindIR_GenOP_SetMinVal() local
[all …]
|
| D | mindir_tensor.cc | 31 …auto ops_offset = schema::CreateTensor(fbb, 0, DataType::DATA_TYPE_INT32, 0, schema::Format::Forma… in MindIR_Tensor_Create() local
44 auto ops_offset = in MindIR_Tensor_Create() local
84 auto ops_offset = schema::CreateTensor( in MindIR_Tensor_SetName() local
132 auto ops_offset = in MindIR_Tensor_SetDataType() local
180 …auto ops_offset = schema::CreateTensor(fbb, 0, value->dataType(), fbb.CreateVector(dims.data(), di… in MindIR_Tensor_SetDims() local
228 …auto ops_offset = schema::CreateTensor(fbb, 0, value->dataType(), dims, static_cast<schema::Format… in MindIR_Tensor_SetFormat() local
275 auto ops_offset = schema::CreateTensor( in MindIR_Tensor_SetData() local
333 auto ops_offset = in MindIR_Tensor_SetQuantParams() local
|