Home
last modified time | relevance | path

Searched refs:primitive_type (Results 1 – 25 of 68) sorted by relevance

123

/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dmulti_conv_info.h27 …explicit MultiConvSplit(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_t…
29 …: MultiNodeSplit(), strategy_(strategy), primitive_type_(primitive_type), fmk_type_(fmk_type), num… in MultiNodeSplit()
73 …MultiConvSplitN(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_type = -1…
74 : MultiConvSplit(strategy, primitive_type, fmk_type, num) {} in MultiConvSplit() argument
84 …MultiConvSplitCIN(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_type = …
85 : MultiConvSplit(strategy, primitive_type, fmk_type, num) {} in MultiConvSplit() argument
95 …MultiConvSplitCOUT(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_type =…
97 : MultiConvSplit(strategy, primitive_type, fmk_type, num) {} in MultiConvSplit() argument
107 …MultiConvSplitH(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_type = -1…
108 : MultiConvSplit(strategy, primitive_type, fmk_type, num) {} in MultiConvSplit() argument
Dmulti_node_split.h39 …explicit MultiNodeSplitProxy(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t …
41 …: MultiNodeSplit(), strategy_(strategy), primitive_type_(primitive_type), fmk_type_(fmk_type), num… in MultiNodeSplit()
/third_party/mindspore/mindspore/lite/src/ops/compat/
Dcompat_register.h40 …void InsertTransferAttrFuncMap(int schema_version, int primitive_type, TransferAttrFunc transfer_a… in InsertTransferAttrFuncMap() argument
41 int key = primitive_type * 10 + schema_version; in InsertTransferAttrFuncMap()
45 TransferAttrFunc GetTransferAttrFunc(int schema_version, int primitive_type) { in GetTransferAttrFunc() argument
46 int key = primitive_type * 10 + schema_version; in GetTransferAttrFunc()
61 Register(int schema_version, int primitive_type, TransferAttrFunc transfer_attr_func) { in Register() argument
62 …CompatRegistry::GetInstance()->InsertTransferAttrFuncMap(schema_version, primitive_type, transfer_… in Register()
/third_party/boost/boost/archive/
Dbasic_archive.hpp249 BOOST_CLASS_IMPLEMENTATION(boost::serialization::library_version_type, primitive_type)
250 BOOST_CLASS_IMPLEMENTATION(boost::archive::version_type, primitive_type)
251 BOOST_CLASS_IMPLEMENTATION(boost::archive::class_id_type, primitive_type)
252 BOOST_CLASS_IMPLEMENTATION(boost::archive::class_id_reference_type, primitive_type)
253 BOOST_CLASS_IMPLEMENTATION(boost::archive::class_id_optional_type, primitive_type)
254 BOOST_CLASS_IMPLEMENTATION(boost::archive::class_name_type, primitive_type)
255 BOOST_CLASS_IMPLEMENTATION(boost::archive::object_id_type, primitive_type)
256 BOOST_CLASS_IMPLEMENTATION(boost::archive::object_reference_type, primitive_type)
257 BOOST_CLASS_IMPLEMENTATION(boost::archive::tracking_type, primitive_type)
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dpopulate_register.h79 Registry(int primitive_type, ParameterGen creator, int version) { in Registry() argument
80 PopulateRegistry::GetInstance()->InsertParameterMap(primitive_type, creator, version); in Registry()
85 #define REG_POPULATE(primitive_type, creator, version) \ argument
86 static Registry g_##primitive_type##version(primitive_type, creator, version);
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dcommon.cc42 auto primitive_type = static_cast<schema::PrimitiveType>(op_parameter->type_); in TestMain() local
86 if (packed_op.count(primitive_type)) { in TestMain()
104 kernel::KernelKey key{kernel::kGPU, kernel_inputs.front()->data_type(), primitive_type}; in TestMain()
110 … std::cerr << "can't get registry function for: " << schema::EnumNamePrimitiveType(primitive_type) in TestMain()
117 …std::cerr << "call registry function error: " << schema::EnumNamePrimitiveType(primitive_type) << … in TestMain()
130 …std::cerr << "call registry function error: " << schema::EnumNamePrimitiveType(primitive_type) << … in TestMain()
134 kernel->set_name(schema::EnumNamesPrimitiveType()[primitive_type]); in TestMain()
191 if (tensor->category() != VAR && packed_op.count(primitive_type)) { in TestMain()
218 auto primitive_type = static_cast<schema::PrimitiveType>(op_parameter->type_); in TestMain() local
251 if (packed_op.count(primitive_type)) { in TestMain()
[all …]
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/
Dop_coder_builder.cc28 int primitive_type = GetPrimitiveType(node_->primitive_, schema_version); in build() local
29 CoderKey coder_key(target_, data_type_, primitive_type); in build()
46 … << mindspore::schema::EnumNamePrimitiveType(static_cast<schema::PrimitiveType>(primitive_type)) in build()
54 if (primitive_type != schema::PrimitiveType_Custom) { in build()
58 op_coder->set_type(primitive_type); in build()
/third_party/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_emit.h96 etna_draw_primitives(struct etna_cmd_stream *stream, uint32_t primitive_type, in etna_draw_primitives() argument
102 etna_cmd_stream_emit(stream, primitive_type); in etna_draw_primitives()
109 uint32_t primitive_type, uint32_t start, in etna_draw_indexed_primitives() argument
115 etna_cmd_stream_emit(stream, primitive_type); in etna_draw_indexed_primitives()
125 uint32_t indexed, uint32_t primitive_type, in etna_draw_instanced() argument
133 VIV_FE_DRAW_INSTANCED_HEADER_TYPE(primitive_type) | in etna_draw_instanced()
/third_party/vk-gl-cts/external/openglcts/modules/gl/
Dgl3cClipDistance.cpp584 glw::GLenum primitive_type = m_primitive_types[i_primitive_type]; in iterate() local
588 glw::GLuint framebuffer_size = (primitive_type == GL_POINTS) ? 1 : 32; in iterate()
613 redeclaration, redeclaration, i_clip_count, i_clip_function, primitive_type); in iterate()
641 gl3cts::ClipDistance::Utility::VertexArrayObject vertex_array_object(gl, primitive_type); in iterate()
644 prepareGeometry(gl, primitive_type); in iterate()
665 if (!checkResults(primitive_type, i_clip_function, results)) in iterate()
669 << glu::getPrimitiveTypeStr(primitive_type) in iterate()
714 glw::GLenum primitive_type) in prepareVertexShaderCode() argument
757 switch (primitive_type) in prepareVertexShaderCode()
781 const glw::Functions& gl, const glw::GLenum primitive_type) in prepareGeometry() argument
[all …]
Dgl3cTransformFeedbackTests.hpp424 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
504 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
538 virtual void draw(glw::GLenum primitive_type) in draw() argument
540 CaptureGeometryInterleaved::draw(primitive_type); in draw()
558 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type) in checkTransformFeedbackBuffer() argument
560 return CaptureVertexSeparate::checkTransformFeedbackBuffer(bind_case, primitive_type); in checkTransformFeedbackBuffer()
705 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
737 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
771 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
806 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
Dgl3cClipDistance.hpp144 VertexArrayObject(const glw::Functions& gl, const glw::GLenum primitive_type); // create empty vao
263 glw::GLuint clip_function, glw::GLenum primitive_type);
266 const glw::GLenum primitive_type);
268 …bool checkResults(glw::GLenum primitive_type, glw::GLuint clip_function, std::vector<glw::GLfloat>…
/third_party/boost/boost/serialization/
Dstring.hpp25 BOOST_CLASS_IMPLEMENTATION(std::string, boost::serialization::primitive_type)
27 BOOST_CLASS_IMPLEMENTATION(std::wstring, boost::serialization::primitive_type)
Dtracking.hpp63 mpl::int_<primitive_type>
112 mpl::int_<primitive_type> \
Dlevel.hpp58 mpl::int_<primitive_type>,
70 mpl::int_<primitive_type>,
/third_party/mindspore/mindspore/lite/examples/converter_extend/src/
Dpass_registry_tutorial.cc29 bool CheckPrimitiveTypeTutorial(const AnfNodePtr &node, const PrimitivePtr &primitive_type) { in CheckPrimitiveTypeTutorial() argument
35 return IsPrimitive(cnode->input(0), primitive_type); in CheckPrimitiveTypeTutorial()
37 return IsPrimitive(node, primitive_type); in CheckPrimitiveTypeTutorial()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Darithmetic_self_fp32.cc32 ArithmeticSelfFunc ArithmeticSelfCPUKernel::GetArithmeticSelfFun(int primitive_type) const { in GetArithmeticSelfFun()
48 if (type_func_table[i].primitive_type_ == primitive_type) { in GetArithmeticSelfFun()
55 ArithmeticSelfBoolFunc ArithmeticSelfCPUKernel::GetArithmeticSelfBoolFun(int primitive_type) const { in GetArithmeticSelfBoolFun()
56 if (primitive_type == mindspore::schema::PrimitiveType_LogicalNot) { in GetArithmeticSelfBoolFun()
Darithmetic_fp32.cc30 auto primitive_type = param_->op_parameter_.type_; in Init() local
31 if (primitive_type == schema::PrimitiveType_Eltwise) { in Init()
34 primitive_type = schema::PrimitiveType_MulFusion; in Init()
37 primitive_type = schema::PrimitiveType_AddFusion; in Init()
40 primitive_type = schema::PrimitiveType_Maximum; in Init()
47 InitRunFunction(primitive_type); in Init()
190 void ArithmeticCPUKernel::InitRunFunction(int primitive_type) { in InitRunFunction() argument
240 …if (fun_table[i].primitive_type_ == primitive_type && fun_table[i].activation_type_ == param_->act… in InitRunFunction()
Darithmetic_self_fp32.h56 ArithmeticSelfFunc GetArithmeticSelfFun(int primitive_type) const;
57 ArithmeticSelfBoolFunc GetArithmeticSelfBoolFun(int primitive_type) const;
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Darithmetic_fp32_coder.cc30 void ArithmeticFP32Coder::InitRunFunction(int primitive_type) { in InitRunFunction() argument
69 if (fun_table[i].primitive_type_ == primitive_type && in InitRunFunction()
254 auto primitive_type = arithmetic_parameter_->op_parameter_.type_; in Prepare() local
255 if (primitive_type == schema::PrimitiveType_Eltwise) { in Prepare()
258 primitive_type = schema::PrimitiveType_MulFusion; in Prepare()
261 primitive_type = schema::PrimitiveType_AddFusion; in Prepare()
264 primitive_type = schema::PrimitiveType_Maximum; in Prepare()
271 InitRunFunction(primitive_type); in Prepare()
/third_party/mindspore/mindspore/lite/src/
Dlite_kernel_util.h38 const schema::PrimitiveType &primitive_type);
39 …ontainsSpecificNode(const kernel::LiteKernel *kernel, const schema::PrimitiveType &primitive_type);
Dlite_kernel_util.cc222 … const schema::PrimitiveType &primitive_type) { in GetInputsSpecificNode() argument
224 if (input->type() == primitive_type) { in GetInputsSpecificNode()
232 const schema::PrimitiveType &primitive_type) { in InputsContainsSpecificNode() argument
233 if (GetInputsSpecificNode(kernel, primitive_type)) { in InputsContainsSpecificNode()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Darithmetic_compare_fp16.cc46 ArithmeticCompareFuncFp16 GetArithmeticCompareFun(int primitive_type, int activation_type) { in GetArithmeticCompareFun() argument
49 if (arithmetic_cp_fun_table_fp16[i].primitive_type_ == primitive_type && in GetArithmeticCompareFun()
57 ArithmeticCompareOptFuncFp16 GetOptimizedArithmeticCompareFun(int primitive_type, int activation_ty… in GetOptimizedArithmeticCompareFun() argument
60 if (arithmetic_cp_fun_table_fp16[i].primitive_type_ == primitive_type && in GetOptimizedArithmeticCompareFun()
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc49 …ckPrimitiveType(const mindspore::AnfNodePtr &node, const mindspore::PrimitivePtr &primitive_type) { in CheckPrimitiveType() argument
55 return IsPrimitive(cnode->input(0), primitive_type); in CheckPrimitiveType()
57 return IsPrimitive(node, primitive_type); in CheckPrimitiveType()
/third_party/mindspore/mindspore/lite/src/common/
Dprim_util.cc56 int GenPrimVersionKey(int primitive_type, int schema_version) { return primitive_type * 1000 + sche… in GenPrimVersionKey() argument
/third_party/mindspore/mindspore/lite/micro/coder/
Dtrain.cc72 int primitive_type = GetPrimitiveType(node->primitive_, schema_version); in TransformGraphForTrain() local
73 auto item = std::find(loss_types.begin(), loss_types.end(), primitive_type); in TransformGraphForTrain()

123