Home
last modified time | relevance | path

Searched refs:getOutputOperandDimensions (Results 1 – 11 of 11) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestExecution.cpp686 EXPECT_EQ(execution.getOutputOperandDimensions(0, &dimensions), WrapperResult::BAD_STATE); in getDimensionsWhileRunning()
779 ASSERT_EQ(execution.getOutputOperandDimensions(0, &dimensions), kExpectResult); in TestWait()
782 ASSERT_EQ(execution.getOutputOperandDimensions(0, &dimensions), in TestWait()
806 ASSERT_EQ(execution.getOutputOperandDimensions(0, &dimensions), kExpectResult); in TestWait()
809 ASSERT_EQ(execution.getOutputOperandDimensions(0, &dimensions), in TestWait()
840 ASSERT_EQ(execution.getOutputOperandDimensions(0, &dimensions), kExpectResult); in TestWait()
843 ASSERT_EQ(execution.getOutputOperandDimensions(0, &dimensions), in TestWait()
877 ASSERT_EQ(execution.getOutputOperandDimensions(0, &dimensions), kExpectResult); in TestWait()
880 ASSERT_EQ(execution.getOutputOperandDimensions(0, &dimensions), in TestWait()
H A DTestGenerated.cpp287 execution.getOutputOperandDimensions(i, &actualDimensions)); in executeInternal()
H A DTestNeuralNetworksWrapper.h535 Result getOutputOperandDimensions(uint32_t index, std::vector<uint32_t>* dimensions) { in getOutputOperandDimensions() function
H A DTestUnspecifiedDimensions.cpp406 ASSERT_EQ(execution.getOutputOperandDimensions(0, &outputShape), result); in TestOne()
H A DSupportLibraryTestGenerated.cpp380 ASSERT_EQ(Result::NO_ERROR, execution.getOutputOperandDimensions(i, &actualDimensions)); in executeWithCompilation()
/aosp12/packages/modules/NeuralNetworks/runtime/
H A DExecutionBuilder.h100 int getOutputOperandDimensions(uint32_t index, uint32_t* dimensions);
H A DExecutionBuilder.cpp479 int ExecutionBuilder::getOutputOperandDimensions(uint32_t index, uint32_t* dimensions) { in getOutputOperandDimensions() function in android::nn::ExecutionBuilder
H A DNeuralNetworks.cpp1312 return r->getOutputOperandDimensions(index, dimensions); in ANeuralNetworksExecution_getOutputOperandDimensions()
/aosp12/packages/modules/NeuralNetworks/shim_and_sl/
H A DShimPreparedModel.cpp295 auto result = execution->getOutputOperandDimensions(i, &outputDims); in executeSynchronously()
/aosp12/packages/modules/NeuralNetworks/runtime/include/
H A DNeuralNetworksWrapper.h687 Result getOutputOperandDimensions(uint32_t index, std::vector<uint32_t>* dimensions) { function
/aosp12/packages/modules/NeuralNetworks/shim_and_sl/include/
H A DSupportLibraryWrapper.h635 Result getOutputOperandDimensions(uint32_t index, std::vector<uint32_t>* dimensions) { in getOutputOperandDimensions() function