Searched refs:ANeuralNetworksModel (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/lite/nnapi/ |
D | nnapi_handler.h | 104 nnapi_->ANeuralNetworksModel_create = [](ANeuralNetworksModel** model) { in ModelCreateReturns() 105 *model = reinterpret_cast<ANeuralNetworksModel*>(1); in ModelCreateReturns() 110 void StubModelCreateWith(int(stub)(ANeuralNetworksModel** model)) { in StubModelCreateWith() 117 [](ANeuralNetworksModel* model, in AddOperandReturns() 121 void StubAddOperandWith(int(stub)(ANeuralNetworksModel* model, in StubAddOperandWith() 129 [](ANeuralNetworksModel* model, int32_t index, const void* buffer, in SetOperandValueReturns() 136 [](ANeuralNetworksModel* model, ANeuralNetworksOperationType type, in AddOperationReturns() 142 int(stub)(ANeuralNetworksModel* model, ANeuralNetworksOperationType type, in StubAddOperationWith() 151 [](ANeuralNetworksModel* model, uint32_t inputCount, in IdentifyInputAndOutputsReturns() 159 [](ANeuralNetworksModel* model, bool allow) { return Value; }; in RelaxComputationFloatReturns() [all …]
|
D | nnapi_implementation.h | 86 int (*ANeuralNetworksModel_create)(ANeuralNetworksModel** model); 99 void (*ANeuralNetworksModel_free)(ANeuralNetworksModel* model); 114 int (*ANeuralNetworksModel_finish)(ANeuralNetworksModel* model); 147 ANeuralNetworksModel* model, const ANeuralNetworksOperandType* type); 172 int (*ANeuralNetworksModel_setOperandValue)(ANeuralNetworksModel* model, 196 ANeuralNetworksModel* model, int32_t index, 225 ANeuralNetworksModel* model, int32_t index, 248 int (*ANeuralNetworksModel_addOperation)(ANeuralNetworksModel* model, 278 ANeuralNetworksModel* model, uint32_t inputCount, const uint32_t* inputs, 306 ANeuralNetworksModel* model, bool allow); [all …]
|
D | NeuralNetworksTypes.h | 318 typedef struct ANeuralNetworksModel ANeuralNetworksModel; typedef 479 typedef int (*ANeuralNetworksModel_create_fn)(ANeuralNetworksModel** model); 481 typedef int (*ANeuralNetworksModel_finish_fn)(ANeuralNetworksModel* model); 483 typedef void (*ANeuralNetworksModel_free_fn)(ANeuralNetworksModel* model); 486 ANeuralNetworksModel* model, ANeuralNetworksCompilation** compilation); 498 ANeuralNetworksModel* model, const ANeuralNetworksOperandType* type); 501 ANeuralNetworksModel* model, int32_t index, const void* buffer, 505 ANeuralNetworksModel* model, int32_t index, 509 ANeuralNetworksModel* model, int32_t index, 513 ANeuralNetworksModel* model, ANeuralNetworksOperationType type, [all …]
|
D | NeuralNetworksShim.h | 154 inline int ANeuralNetworksModel_create(ANeuralNetworksModel** model) { in ANeuralNetworksModel_create() 170 inline void ANeuralNetworksModel_free(ANeuralNetworksModel* model) { in ANeuralNetworksModel_free() 188 inline int ANeuralNetworksModel_finish(ANeuralNetworksModel* model) { in ANeuralNetworksModel_finish() 223 ANeuralNetworksModel* model, const ANeuralNetworksOperandType* type) { in ANeuralNetworksModel_addOperand() 251 inline int ANeuralNetworksModel_setOperandValue(ANeuralNetworksModel* model, in ANeuralNetworksModel_setOperandValue() 279 ANeuralNetworksModel* model, int32_t index, in ANeuralNetworksModel_setOperandSymmPerChannelQuantParams() 310 ANeuralNetworksModel* model, int32_t index, in ANeuralNetworksModel_setOperandValueFromMemory() 336 inline int ANeuralNetworksModel_addOperation(ANeuralNetworksModel* model, in ANeuralNetworksModel_addOperation() 369 ANeuralNetworksModel* model, uint32_t inputCount, const uint32_t* inputs, in ANeuralNetworksModel_identifyInputsAndOutputs() 398 ANeuralNetworksModel* model, bool allow) { in ANeuralNetworksModel_relaxComputationFloat32toFloat16() [all …]
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_device_selection_test.cc | 100 [](const ANeuralNetworksModel* model, in SetUp() 119 [](ANeuralNetworksModel* model, in TEST_F() 136 [](ANeuralNetworksModel* model, in TEST_F() 160 [](ANeuralNetworksModel* model, in TEST_F() 289 [](const ANeuralNetworksModel* model, in TEST_F() 324 [](const ANeuralNetworksModel* model, in TEST_F() 429 [](const ANeuralNetworksModel* model, in TEST_F() 439 nnapi_mock_->StubModelCreateWith([](ANeuralNetworksModel** model) -> int { in TEST_F() 441 *model = reinterpret_cast<ANeuralNetworksModel*>(1); in TEST_F() 446 [](ANeuralNetworksModel* model, ANeuralNetworksOperationType type, in TEST_F() [all …]
|
D | nnapi_delegate_kernel.h | 125 void operator()(ANeuralNetworksModel* model) { in operator() 323 std::unique_ptr<ANeuralNetworksModel, NNFreeModel> nn_model_;
|
D | nnapi_delegate_mock_test.h | 49 nnapi_->ANeuralNetworksModel_free = [](ANeuralNetworksModel* model) {}; in NnApiHandler()
|
D | nnapi_delegate_nnapi_failure_handling_test.cc | 113 [](ANeuralNetworksModel* model, ANeuralNetworksOperationType type, in TEST_F()
|
D | nnapi_delegate_signed_quantization_test.cc | 134 [](ANeuralNetworksModel* model, in SetUp()
|
D | nnapi_delegate.cc | 717 ANeuralNetworksModel* nn_model, int* nnapi_errno, in NNAPIOpBuilder() 1502 ANeuralNetworksModel* const nn_model_; 3642 ANeuralNetworksModel* model = nullptr; in Init()
|
/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/ |
D | nnapi_plugin_test.cc | 66 [](const ANeuralNetworksModel* model, in SetUp() 185 [](const ANeuralNetworksModel* model, in TEST_F() 201 [](const ANeuralNetworksModel* model, in TEST_F()
|