Home
last modified time | relevance | path

Searched refs:m_valueDim (Results 1 – 2 of 2) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dconstant_of_shape_test.cpp46 std::vector<int32_t> m_valueDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::ConstantOfShapeBuilderTest
83 SaveValue(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_CONSTANT_OF_SHAPE_VALUE);
99 SaveValue(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_CONSTANT_OF_SHAPE_VALUE);
120 SaveValue(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_CONSTANT_OF_SHAPE_VALUE);
139 SaveValue(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_CONSTANT_OF_SHAPE_VALUE);
184 SaveValue(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_CONSTANT_OF_SHAPE_VALUE);
201 std::shared_ptr<NNTensor> valueTensor = TransToNNTensor(OH_NN_INT64, m_valueDim,
223 SaveValue(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_CONSTANT_OF_SHAPE_VALUE);
239 SaveValue(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
258 SaveValue(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_CONSTANT_OF_SHAPE_VALUE);
[all …]
H A Dsparse_to_dense_test.cpp42 std::vector<int32_t> m_valueDim {2}; member in OHOS::NeuralNetworkRuntime::UnitTest::SparseToDenseBuilderTest
59 valueTensor = TransToNNTensor(OH_NN_FLOAT32, m_valueDim, nullptr, OH_NN_TENSOR); in SetInputTensor()