Searched refs:StridedSliceBuilder (Results 1 – 3 of 3) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | strided_slice_builder.cpp | 30 StridedSliceBuilder::StridedSliceBuilder() {} in StridedSliceBuilder() function in OHOS::NeuralNetworkRuntime::Ops::StridedSliceBuilder
32 StridedSliceBuilder::~StridedSliceBuilder() {} in ~StridedSliceBuilder()
34 OH_NN_ReturnCode StridedSliceBuilder::SetInputOutput(const std::vector<uint32_t>& inputsIndex, in SetInputOutput()
50 OH_NN_ReturnCode StridedSliceBuilder::SetBeginMask(const std::shared_ptr<NNTensor>& tensor) in SetBeginMask()
67 OH_NN_ReturnCode StridedSliceBuilder::SetEndMask(const std::shared_ptr<NNTensor>& tensor) in SetEndMask()
84 OH_NN_ReturnCode StridedSliceBuilder::SetEllipsisMask(const std::shared_ptr<NNTensor>& tensor) in SetEllipsisMask()
101 OH_NN_ReturnCode StridedSliceBuilder::SetNewAxisMask(const std::shared_ptr<NNTensor>& tensor) in SetNewAxisMask()
118 OH_NN_ReturnCode StridedSliceBuilder::SetShrinkAxisMask(const std::shared_ptr<NNTensor>& tensor) in SetShrinkAxisMask()
141 OH_NN_ReturnCode StridedSliceBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()
184 LiteGraphPrimitvePtr StridedSliceBuilder::GetPrimitive() in GetPrimitive()
[all …]
|
| H A D | strided_slice_builder.h | 25 class StridedSliceBuilder : public OpsBuilder {
27 typedef OH_NN_ReturnCode (StridedSliceBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
29 StridedSliceBuilder();
30 ~StridedSliceBuilder() override;
55 {OH_NN_STRIDED_SLICE_BEGIN_MASK, &StridedSliceBuilder::SetBeginMask},
56 {OH_NN_STRIDED_SLICE_END_MASK, &StridedSliceBuilder::SetEndMask},
57 {OH_NN_STRIDED_SLICE_ELLIPSIS_MASK, &StridedSliceBuilder::SetEllipsisMask},
58 {OH_NN_STRIDED_SLICE_NEW_AXIS_MASK, &StridedSliceBuilder::SetNewAxisMask},
59 {OH_NN_STRIDED_SLICE_SHRINK_AXIS_MASK, &StridedSliceBuilder::SetShrinkAxisMask}
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | strided_slice_builder_test.cpp | 46 StridedSliceBuilder m_builder;
|