- 10 12月, 2019 1 次提交
-
-
由 Wilber 提交于
修改了选kernel的逻辑,默认从模型文件中读取出lod_tensor的data type,在static_kernel_pick pass中如果kernel输入输出的类型与读取的data type完全一致,则选择该Kernel的概率增大。 - 增加 从模型文件__model__读取lod_tensor的data type到cpp::vardesc - program中增加unordered_map<string, type>字段,并在 Program::PrepareWorkspace中对该字段赋值 - 修改了node.h文件,将const Type* 更改为Type*,并在SSAGraph::Build过程中为符合条件的type*赋值 - static_kernel_pick_pass中添加新规则,如果kernel的输入类型输出类型与__model__中存储的类型的一致,则score*=2。 - 支持模型中用到sequence_reverse_float kernel(输入输出均为float)和sequence_reverse_int64 kernel(输入输出均为int64),能够根据输入输出type选kernel
-
- 04 12月, 2019 1 次提交
-
-
由 石晓伟 提交于
-
- 24 10月, 2019 1 次提交
-
-
由 liu zhengxi 提交于
* make inceptionv4, resnet50, googlenet can run on x86 paltform and fix the compare part in x86 unittests, test=develop * fix googlenet tests for benchmark record, test=develop * [framework][profile] fix profile dump bug when op is feed and fetch test=develop (sangoly)
-
- 27 9月, 2019 1 次提交
-
-
由 sangoly 提交于
-
- 30 8月, 2019 1 次提交
-
-
由 Zhen Wang 提交于
* Add precision and persistable attrs for the tensor. And fix cxx light and full api demo. * update precision2string methods. test=develop * move the save logic to the front of the run in mobilenetv1_full_api.cc, test=develop. * add comments for UpdateVarsOfProgram. test=develop
-
- 16 8月, 2019 1 次提交
-
-
由 Yan Chunwei 提交于
-