Home
last modified time | relevance | path

Searched defs:keepDimsTensor (Results 1 – 8 of 8) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dall_test.cpp53 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
174 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
209 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
H A Dreduce_all_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, variable
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
H A Dreduce_l2_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, variable
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
H A Dreduce_max_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, variable
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
H A Dreduce_mean_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
H A Dreduce_min_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, variable
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
H A Dreduce_prod_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
H A Dreduce_sum_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, variable
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable