Searched refs:nn_type (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 352 TfLiteStatus AddScalarOperand(T value, int32_t nn_type) { in AddScalarOperand() argument
353 ANeuralNetworksOperandType operand_type{.type = nn_type}; in AddScalarOperand()
367 int32_t nn_type) { in AddVectorOperand() argument
369 .type = nn_type, .dimensionCount = 1, .dimensions = &num_values}; in AddVectorOperand()
415 int32_t nn_type = 0; in AddTensor() local
432 nn_type = ANEURALNETWORKS_TENSOR_FLOAT32; in AddTensor()
435 nn_type = ANEURALNETWORKS_TENSOR_QUANT8_ASYMM; in AddTensor()
444 nn_type = ANEURALNETWORKS_TENSOR_QUANT8_SYMM; in AddTensor()
448 nn_type = ANEURALNETWORKS_TENSOR_INT32; in AddTensor()
458 nn_type, static_cast<uint32_t>(tensor->dims->size), in AddTensor()
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | nnapi_delegate.cc | 156 int32_t nn_type = 0; in addTensorOperands() local
167 nn_type = ANEURALNETWORKS_TENSOR_FLOAT32; in addTensorOperands()
170 nn_type = ANEURALNETWORKS_TENSOR_QUANT8_ASYMM; in addTensorOperands()
179 nn_type = ANEURALNETWORKS_TENSOR_INT32; in addTensorOperands()
202 nn_type, static_cast<uint32_t>(tensor->dims->size), in addTensorOperands()
435 &next_id](int nn_type) { in AddOpsAndParams() argument
439 ANeuralNetworksOperandType operand_type{nn_type, 2, dim.data(), 0, 0}; in AddOpsAndParams()
|
| /external/tensorflow/tensorflow/lite/ |
| D | nnapi_delegate.cc | 127 int32_t nn_type = 0; in addTensorOperands() local
138 nn_type = ANEURALNETWORKS_TENSOR_FLOAT32; in addTensorOperands()
141 nn_type = ANEURALNETWORKS_TENSOR_QUANT8_ASYMM; in addTensorOperands()
146 nn_type = ANEURALNETWORKS_TENSOR_INT32; in addTensorOperands()
170 nn_type, static_cast<uint32_t>(tensor->dims->size), in addTensorOperands()
405 &next_id](int nn_type) { in AddOpsAndParams() argument
409 ANeuralNetworksOperandType operand_type{nn_type, 2, dim.data(), 0, 0}; in AddOpsAndParams()
|