Searched defs:keepDimsTensor (Results 1 – 3 of 3) sorted by relevance
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| D | reduce_prod_builder_test.cpp | 53 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
162 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
185 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
221 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
|
| D | reduce_all_builder_test.cpp | 53 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
162 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
185 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDim, variable
221 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
|
| D | reduce_mean_builder_test.cpp | 53 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
162 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
185 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
221 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
|