onnx-mlir

Commit Graph

Author	SHA1	Message	Date
Tung D. Le	e89e51699b	Lowering softmax (#14 ) * Rebase * Use max normalization * Handle axis * Add tests * Update SharingWork.md * Remove redundant spaces * Format code * Rebase * Change from the use of Value* to Value * Add end-to-end tests Co-authored-by: Tian Jin <tjingrant@gmail.com>	2020-01-20 21:57:32 -05:00
TUNG LEDUC	06a968d4a1	[MLIR] Add broadcasting support for element wise operations (#398 ) * Add broadcasting support for elementwise operations * Remove MLIRDialect from MLIRWholeArchiveLibs * Rewrite getLoopIVsForBroadcasting * Compute dimensions for allocating result memory * Compute dimensions for allocating result memory (revised) * Use static dimension for element-wise operation testcases * Add a test for addition with broadcasting * Missed Traits.h when merging * Revise * Update SharedWork.md * Broadcasting for variadic operations * Edit comments * Update SharedWork.md * Reorganize the code * Add CHECK-LABEL for test_add_with_broadcasting	2019-12-21 02:08:27 -05:00
TUNG LEDUC	5ed79083d5	[MLIR] Add support for Max, Min, Sum, Elu, Selu, LeakyRelu, HardSigmoid (#395 ) * Lower ONNXSumOp * Add inferShapes() and test cases * Load the first operand to the result * Update SharingWork.md * Update SharingWork.md * Update SharingWork.md * Add support for Max, Min * Pass operation instead of location to mapToLowerScalarOp * Add support for Elu, Selu, LeakyRelu, HardSigmoid * Add test cases * Update SharingWork.md * Rewrite the part of lowering variadic ops and use it for binary ops * Use two diffenrent templates for Unary and Variadic Ops * Revise the code	2019-12-21 02:02:09 -05:00
Alexandre E Eichenberger	fb1b43f842	Create SharingWork.md (#394 ) Add tables to keep track of the work, initially with a section on ONNX to KRNL work. Add more sections as you see fit	2019-12-21 02:01:47 -05:00

4 Commits