Skip to content

Commit

Permalink
Fixed typing error in gather test
Browse files Browse the repository at this point in the history
Deleted the external output tensor creation in scalar_1d test

Type: Bug Fix
Signed-off-by: Feiyue Chen <Feiyue.Chen@verisilicon.com>
  • Loading branch information
chenfeiyue-cfy committed May 15, 2024
1 parent e1c2f0a commit 0483a6d
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions src/tim/vx/ops/gather_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -168,14 +168,11 @@ TEST(Gather, scalar_index_input1D) {
tim::vx::TensorAttribute::INPUT);
tim::vx::TensorSpec gatherout_spec(tim::vx::DataType::FLOAT32, out_shape,
tim::vx::TensorAttribute::OUTPUT);
tim::vx::TensorSpec output_spec(tim::vx::DataType::FLOAT32, out_shape,
tim::vx::TensorAttribute::OUTPUT);

auto input_tensor = graph->CreateTensor(input_spec);
auto index_tensor = graph->CreateTensor(index_spec);
index_tensor->SetScalar(1);
auto gatherout_tensor = graph->CreateTensor(gatherout_spec);
auto output_tensor = graph->CreateTensor(output_spec);

std::vector<float> in_data = {
1,2,3,4,5};
Expand Down

0 comments on commit 0483a6d

Please sign in to comment.