Searched refs:m_axisDim (Results 1 – 1 of 1) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | crop_test.cpp | 47 std::vector<int32_t> m_axisDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::CropBuilderTest
94 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS);
110 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS);
134 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS);
153 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS);
194 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_axisDim,
215 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS);
237 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
253 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS);
289 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS);
[all …]
|