Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
193 changes: 0 additions & 193 deletions bridge/lib/Bridge.chpl

This file was deleted.

20 changes: 8 additions & 12 deletions bridge/lib/bridge.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -685,7 +685,7 @@ extern "C" bridge_tensor_t softplus(
.beta(beta)
.threshold(threshold));

return torch_to_bridge(t_output)
return torch_to_bridge(t_output);
}


Expand All @@ -695,11 +695,7 @@ extern "C" bridge_tensor_t threshold(
float value
) {
auto t_input = bridge_to_torch(input);
auto t_output = tnf::threshold(t_input,
tnf::ThresholdFuncOptions()
.threshold(threshold)
.value(value));

auto t_output = tnf::threshold(t_input,tnf::ThresholdFuncOptions(threshold,value));
return torch_to_bridge(t_output);
}

Expand All @@ -713,7 +709,7 @@ extern "C" bridge_tensor_t celu(
tnf::CELUFuncOptions()
.alpha(alpha));

return t_output;
return torch_to_bridge(t_output);
}


Expand Down Expand Up @@ -772,7 +768,7 @@ extern "C" bridge_tensor_t dropout(
bool training
) {
auto t_input = bridge_to_torch(input);
auto t_output = tnf::dropout(
auto t_output = tnf::dropout(t_input,
tnf::DropoutFuncOptions()
.p(p)
.training(training));
Expand All @@ -787,7 +783,7 @@ extern "C" bridge_tensor_t alpha_dropout(
bool training
) {
auto t_input = bridge_to_torch(input);
auto t_output = tnf::alpha_dropout(
auto t_output = tnf::alpha_dropout(t_input,
tnf::AlphaDropoutFuncOptions()
.p(p)
.training(training));
Expand All @@ -802,7 +798,7 @@ extern "C" bridge_tensor_t feature_alpha_dropout(
bool training
) {
auto t_input = bridge_to_torch(input);
auto t_output = tnf::feature_alpha_dropout(
auto t_output = tnf::feature_alpha_dropout(t_input,
tnf::FeatureAlphaDropoutFuncOptions()
.p(p)
.training(training));
Expand All @@ -817,7 +813,7 @@ extern "C" bridge_tensor_t dropout2d(
bool training
) {
auto t_input = bridge_to_torch(input);
auto t_output = tnf::dropout2d(
auto t_output = tnf::dropout2d(t_input,
tnf::Dropout2dFuncOptions()
.p(p)
.training(training));
Expand All @@ -832,7 +828,7 @@ extern "C" bridge_tensor_t dropout3d(
bool training
) {
auto t_input = bridge_to_torch(input);
auto t_output = tnf::dropout3d(
auto t_output = tnf::dropout3d(t_input,
tnf::Dropout3dFuncOptions()
.p(p)
.training(training));
Expand Down
2 changes: 1 addition & 1 deletion lib/Bridge.chpl
Original file line number Diff line number Diff line change
Expand Up @@ -263,7 +263,7 @@ module Bridge {
) : bridge_tensor_t;

extern proc softmax(
in input : bridge tensor_t,
in input : bridge_tensor_t,
param dim : int(64)
) : bridge_tensor_t;

Expand Down
2 changes: 1 addition & 1 deletion lib/NDArray.chpl
Original file line number Diff line number Diff line change
Expand Up @@ -2600,7 +2600,7 @@ proc ndarray.dropout(p : real = 0.5, training : bool = false) {
:rtype: ndarray(rank, eltType)
*/
proc ndarray.alphaDropout(p : real = 0.5, training : bool = false) {
return Bridge.alphaDroput(
return Bridge.alphaDropout(
this : Bridge.tensorHandle(eltType),
p, training
) : ndarray(rank, eltType);
Expand Down
Loading