Skip to content

N
ngraph
Switch branch/tag
History Find file
  • Amy Zhuang's avatar
    [MLIR] Use call back for MatMul. (#3838) · c737a573
    Amy Zhuang authored 
    * [MLIR] Use call back for MatMul.

* Use callback for Gemm.

* Use mkldnn callback for Softmax.

* Address PR feedback.

* Fix merge errors.

* Change to tail allocation struct.

* Use mkldnn callback for AvgPool.

* Add callbacks for AvgPoolBackprop, MaxPool, and MaxPoolBackprop.

* Fix merge errors.

* Use UnrankedMemRefType for callbacks.

* Address PR feedback.

* Cleanup.

* Address PR feedback.

* Fix a bug.

* Use global variable to hold attributes.

* Convert layout if needed for pooling.

* Address PR feedback.

* Add header.

* Address PR feedback.

* Update Copyright to 2017-2020.

* Address PR feedback.
Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
    c737a573
Name
Last commit
 Last update
..
abc.in.cpp Loading commit data...
abs.in.cpp Loading commit data...
acos.in.cpp Loading commit data...
add.in.cpp Loading commit data...
aliased_output.in.cpp Loading commit data...
all.in.cpp Loading commit data...
any.in.cpp Loading commit data...
api.in.cpp Loading commit data...
arg_reduce.in.cpp Loading commit data...
asin.in.cpp Loading commit data...
atan.in.cpp Loading commit data...
atan2.in.cpp Loading commit data...
auto_broadcast.in.cpp Loading commit data...
autodiff.in.cpp Loading commit data...
batch_mat_mul.in.cpp Loading commit data...
batch_norm.in.cpp Loading commit data...
broadcast.in.cpp Loading commit data...
builder_flatten.in.cpp Loading commit data...
ceiling.in.cpp Loading commit data...
comparison.in.cpp Loading commit data...
computation_reuse.in.cpp Loading commit data...
concat.in.cpp Loading commit data...
constant.in.cpp Loading commit data...
convert.in.cpp Loading commit data...
convolution.in.cpp Loading commit data...
convolution_reference.in.cpp Loading commit data...
cos.in.cpp Loading commit data...
cosh.in.cpp Loading commit data...
cum_sum.in.cpp Loading commit data...
distributed.in.cpp Loading commit data...
divide.in.cpp Loading commit data...
dot.in.cpp Loading commit data...
dyn_broadcast.in.cpp Loading commit data...
dyn_replace_slice_reference.in.cpp Loading commit data...
dyn_reshape.in.cpp Loading commit data...
dyn_slice_reference.in.cpp Loading commit data...
dynamic.in.cpp Loading commit data...
embedding_lookup.in.cpp Loading commit data...
erf.in.cpp Loading commit data...
exp.in.cpp Loading commit data...
floor.in.cpp Loading commit data...
function_name.in.cpp Loading commit data...
fused_op.in.cpp Loading commit data...
gather.in.cpp Loading commit data...
gelu.in.cpp Loading commit data...
generate_mask.in.cpp Loading commit data...
graph_comparison.in.cpp Loading commit data...
group_convolution.in.cpp Loading commit data...
layer_norm.in.cpp Loading commit data...
log.in.cpp Loading commit data...
logical_and.in.cpp Loading commit data...
logical_or.in.cpp Loading commit data...
logical_xor.in.cpp Loading commit data...
lrn.in.cpp Loading commit data...
matmul.in.cpp Loading commit data...
max.in.cpp Loading commit data...
maximum.in.cpp Loading commit data...
min.in.cpp Loading commit data...
minimum.in.cpp Loading commit data...
mlir.in.cpp Loading commit data...
multiple_backends.in.cpp Loading commit data...
multiple_result.in.cpp Loading commit data...
multiply.in.cpp Loading commit data...
negative.in.cpp Loading commit data...
node_name.in.cpp Loading commit data...
not.in.cpp Loading commit data...
numeric.in.cpp Loading commit data...
one_hot.in.cpp Loading commit data...
pad.in.cpp Loading commit data...
parameter_as_output.in.cpp Loading commit data...
partial_slice.in.cpp Loading commit data...
pool.in.cpp Loading commit data...
power.in.cpp Loading commit data...
product.in.cpp Loading commit data...
quantize_dequantize.in.cpp Loading commit data...
quantized_convolution.in.cpp Loading commit data...
quantized_dot.in.cpp Loading commit data...
random_uniform.in.cpp Loading commit data...
range.in.cpp Loading commit data...
relu.in.cpp Loading commit data...
replace_slice.in.cpp Loading commit data...
reshape.in.cpp Loading commit data...
reverse.in.cpp Loading commit data...
reverse_sequence.in.cpp Loading commit data...
round.in.cpp Loading commit data...
scatter.in.cpp Loading commit data...
select.in.cpp Loading commit data...
shape_of.in.cpp Loading commit data...
sigmoid.in.cpp Loading commit data...
sign.in.cpp Loading commit data...
sin.in.cpp Loading commit data...
sinh.in.cpp Loading commit data...
slice.in.cpp Loading commit data...
softmax.in.cpp Loading commit data...
sqrt.in.cpp Loading commit data...
strided_slice.in.cpp Loading commit data...
subtract.in.cpp Loading commit data...
sum.in.cpp Loading commit data...
tan.in.cpp Loading commit data...
tanh.in.cpp Loading commit data...
tile.in.cpp Loading commit data...
topk.in.cpp Loading commit data...
transpose.in.cpp Loading commit data...
unhandled_op.in.cpp Loading commit data...
validate_call.in.cpp Loading commit data...
zero_sized.in.cpp Loading commit data...