Home
last modified time | relevance | path

Searched refs:nn_model (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/
Dnnapi_delegate.cc146 ANeuralNetworksModel* nn_model, in addTensorOperands() argument
205 ANeuralNetworksModel_addOperand(nn_model, &operand_type)); in addTensorOperands()
213 nn_model, next_id, alloc->memory(), in addTensorOperands()
217 nn_model, next_id, tensor->data.raw, tensor->bytes)); in addTensorOperands()
222 ANeuralNetworksModel_setOperandValue(nn_model, next_id, nullptr, 0)); in addTensorOperands()
247 tflite::Interpreter* interpreter, ANeuralNetworksModel* nn_model, in AddOpsAndParams() argument
265 auto add_scalar_int32 = [&nn_model, &augmented_inputs, in AddOpsAndParams()
268 CHECK_NN(ANeuralNetworksModel_addOperand(nn_model, &operand_type)) in AddOpsAndParams()
269 CHECK_NN(ANeuralNetworksModel_setOperandValue(nn_model, next_id, &value, in AddOpsAndParams()
274 auto add_scalar_float32 = [&nn_model, &augmented_inputs, in AddOpsAndParams()
[all …]
/external/tensorflow/tensorflow/lite/
Dnnapi_delegate.cc116 ANeuralNetworksModel* nn_model, in addTensorOperands() argument
173 nnapi->ANeuralNetworksModel_addOperand(nn_model, &operand_type)); in addTensorOperands()
181 nn_model, next_id, alloc->memory(), in addTensorOperands()
185 nn_model, next_id, tensor->data.raw, tensor->bytes)); in addTensorOperands()
190 nn_model, next_id, nullptr, 0)); in addTensorOperands()
215 tflite::Subgraph* subgraph, ANeuralNetworksModel* nn_model, in AddOpsAndParams() argument
234 auto add_scalar_int32 = [nnapi, &nn_model, &augmented_inputs, in AddOpsAndParams()
237 CHECK_NN(nnapi->ANeuralNetworksModel_addOperand(nn_model, &operand_type)) in AddOpsAndParams()
239 nn_model, next_id, &value, sizeof(int32_t))) in AddOpsAndParams()
243 auto add_scalar_float32 = [nnapi, &nn_model, &augmented_inputs, in AddOpsAndParams()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc236 ANeuralNetworksModel* nn_model) in NNAPIOpBuilder() argument
241 nn_model_(nn_model) {} in NNAPIOpBuilder()