Skip to content

N
ngraph
Switch branch/tag
History Find file
  • Pruthvi's avatar
    Implementation of CrossEntropy and CrossEntropyBackprop as fused Op's (#3818) · a1a8a7e3
    Pruthvi authored 
    * - Implementaion of CrossEntropy and CrossEntropyBackprop as fused Op's

* - unit test case for CE fprop
- fix bug in decompose_op

* WIP debug PDPD unit test failure

* fixed broadcasting issue

* -fix bdcast issue for multi dim tensor

* utilities to restore the original tensor shape

* i) style-fix ii) rename variables

* - unit test for multiple dimensions ii) refactor create_mask to seperate function

* - fixed unit tests

* fix style

* set output element type to dynamic in pre_validate and infer shape

* disable ce with one hot unit test on PlaidML

* add CE op to fused_op_tbl

* - add serialzier support for CE and CE Backprop
    a1a8a7e3
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...
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...
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...
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...
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...
tensorview_custom_mem.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...