Home
last modified time | relevance | path

Searched refs:nn_model_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc241 nn_model_(nn_model) {} in NNAPIOpBuilder()
313 nnapi_->ANeuralNetworksModel_addOperand(nn_model_, &operand_type)); in AddDequantize()
322 nn_model_, ANEURALNETWORKS_DEQUANTIZE, 1, in AddDequantize()
341 nn_model_, type, static_cast<uint32_t>(augmented_inputs_.size()), in FinalizeAddOperation()
356 nnapi_->ANeuralNetworksModel_addOperand(nn_model_, &operand_type)); in AddScalarOperand()
360 nn_model_, ann_index, &value, sizeof(T))); in AddScalarOperand()
373 nnapi_->ANeuralNetworksModel_addOperand(nn_model_, &operand_type)); in AddVectorOperand()
378 nn_model_, ann_index, values, sizeof(T) * num_values)); in AddVectorOperand()
393 nnapi_->ANeuralNetworksModel_addOperand(nn_model_, &operand_type)); in AddFloat32OutputTensor()
462 nnapi_->ANeuralNetworksModel_addOperand(nn_model_, &operand_type)); in AddTensor()
[all …]
/external/tensorflow/tensorflow/contrib/lite/
Dnnapi_delegate.cc136 if (nn_model_) { in ~NNAPIDelegate()
137 ANeuralNetworksModel_free(nn_model_); in ~NNAPIDelegate()
138 nn_model_ = nullptr; in ~NNAPIDelegate()
711 if (nn_model_ && nn_compiled_model_) return model_status_; in BuildGraph()
714 if (!nn_model_) { in BuildGraph()
715 CHECK_NN(ANeuralNetworksModel_create(&nn_model_)); in BuildGraph()
746 interpreter, nn_model_, &next_id, &tensor_id_to_nnapi_id)); in BuildGraph()
748 AddOpsAndParams(interpreter, nn_model_, next_id, &model_states_inputs_, in BuildGraph()
767 nn_model_, static_cast<uint32_t>(augmented_inputs.size()), in BuildGraph()
774 nn_model_, interpreter->GetAllowFp16PrecisionForFp32())); in BuildGraph()
[all …]
/external/tensorflow/tensorflow/lite/
Dnnapi_delegate.cc106 if (nn_model_) { in ~NNAPIDelegate()
107 NnApiImplementation()->ANeuralNetworksModel_free(nn_model_); in ~NNAPIDelegate()
108 nn_model_ = nullptr; in ~NNAPIDelegate()
699 if (nn_model_ && nn_compiled_model_) return model_status_; in BuildGraph()
703 if (!nn_model_) { in BuildGraph()
704 CHECK_NN(nnapi->ANeuralNetworksModel_create(&nn_model_)); in BuildGraph()
733 subgraph, nn_model_, &next_id, &tensor_id_to_nnapi_id)); in BuildGraph()
735 AddOpsAndParams(subgraph, nn_model_, next_id, &model_states_inputs_, in BuildGraph()
752 nn_model_, static_cast<uint32_t>(augmented_inputs.size()), in BuildGraph()
759 nn_model_, subgraph->GetAllowFp16PrecisionForFp32())); in BuildGraph()
[all …]
Dnnapi_delegate_disabled.cc32 UNUSED_MEMBER(nn_model_); in ~NNAPIDelegate()
Dnnapi_delegate.h64 ANeuralNetworksModel* nn_model_ = nullptr;