- 22 7月, 2020 1 次提交
-
-
由 hong19860320 提交于
* [Core][ARM] Fix beam_search, eltwise_mul supports broadcast and int64_t data type, add print op and kernel, add exeception test=develop * Fix the dims of parent idx of the arm kernel of beam_search op * elementwise_mul supports int64_t data type with broadcasting * Add print op and kernel for debugging * Support throwing the exception when the internal error occurs * Refine while and conditional_block op kernel * Support the graph optimization on subblocks * Pass program_desc and block_idx into the kernel of the control flow ops(while/conditional_block/subgraph), and create the RuntimeProgram online, it make it possiable to call the control flow ops recursively *Add unit test for masked transformer model
-
- 07 7月, 2020 1 次提交
-
-
由 ysh329 提交于
* fix ttfnet miscs. test=develop
-
- 13 4月, 2020 1 次提交
-
-
由 cc 提交于
-
- 09 3月, 2020 1 次提交
-
-
由 zhupengyang 提交于
* enhance gather, lookup_table arm kernel uts * enhance beam_search, beam_search_decoder, increment rigster
-
- 18 12月, 2019 1 次提交
-
-
由 juncaipeng 提交于
* Support Mask RCNN2 (#2588)
-
- 22 10月, 2019 1 次提交
-
-
由 TianXiaogang 提交于
* feat: add beam_search_special function for support nlp model * fix: add beam_search_compute kernel input and output * feat: add assign op & copy_compute kernel * feat: add fill_const_batch_size_like op & kernel * feat: add layer_norm op and kernel and ut * fix: fix some bugs fix mul_op infer_shape bug when x_dim_idx = 2, x_dims.size()=3 & y_dim_idx = 1, y_dims.size()=2 fix elementwise_compute bug when y axis is all 1 fix beam_search choose math_func wrong bug fix layer_norm get attr bug fix fill_constant_batch_size_like shape_set bug * feat: add gather op and kernel & and transform ut * feats: add ops and fix bugs to support transformer op fix type_cast passes to skip `while` fix elementwise infer_shape bug when x.dims=3 and y.dims={1} & axis=0 fix lookup_table compute bug fix read_from_array/beam_search/increment/compate/gather ops data_type problems * fix: transfomer ut add word read inferface fix copy/gather/norm/layer_norm include path problem * fix:debug info * fix: fix input reshape bug * fix: fix norm bug * style: style fix & test=develop * style: fix operators cmakelist * style: fix operators cmakelist; test=develop * fix and test=develop * fix and test=develop * style: style fix; test=develop
-
- 03 9月, 2019 1 次提交
-
-
由 huzhiqiang 提交于
-
- 16 8月, 2019 1 次提交
-
-
由 Yan Chunwei 提交于
-