Skip to content
Snippets Groups Projects

Vit operators

Merged Houssem ROUIS requested to merge hrouis/aidge_core:vit_operators into tiling
  • Added Erf, Concat, Gather, ReduceMean, Reshape, Slice and transpose operators and their python bindings
  • Modified Softmax operator by adding axis attribute to support import from ONNX
  • Added RemoveDropout Recipe to remove dropout nodes from graphs
Edited by Houssem ROUIS

Merge request reports

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
20 20 def tearDown(self):
21 21 pass
22 22
23 def test_remove_dropout(self):
24 graph_view = aidge_core.sequential([
25 aidge_core.GenericOperator("Conv", 1, 1, 1, "Conv0");
26 aidge_core.GenericOperator("Dropout", 1, 1, 1, name="Dropout0")
  • 66 return std::make_shared<Slice_Op>(*this);
    67 }
    68
    69 void associateInput(const IOIndex_t inputIdx, std::shared_ptr<Data> data) override final {
    70 assert(inputIdx < 4 && "Slice operator supports only 4 inputs");
    71 assert(strcmp(data->type(), Tensor::Type)==0 && "input data must be of Tensor type");
    72 mInputs[inputIdx] = std::dynamic_pointer_cast<Tensor>(data);
    73 }
    74
    75 void computeOutputDims() override final {
    76 if (!mInputs[0]->empty() && !mInputs[1]->empty() && !mInputs[2]->empty()&& !mInputs[3]->empty())
    77 {
    78 DimSize_t nbAxes = mInputs[1]->dims()[0];
    79 const int* axes = static_cast<const int*>(mInputs[1]->getImpl()->rawPtr());
    80 const int* starts = static_cast<const int*>(mInputs[2]->getImpl()->rawPtr());
    81 const int* ends = static_cast<const int*>(mInputs[3]->getImpl()->rawPtr());
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Please register or sign in to reply
    Loading