Searched refs:tensor_flags (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 785 int tensor_flags = 0) { in AddTensorInput() argument
786 return AddTensor(tensor_index, hybrid_op, &augmented_inputs_, tensor_flags); in AddTensorInput()
789 TfLiteStatus AddTensorOutput(int tensor_index, int tensor_flags = 0) { in AddTensorOutput() argument
791 tensor_flags); in AddTensorOutput()
868 int tensor_flags = 0; in TransformHardSwishIntoSupportedOps() local
870 tensor_flags = tensor_flags | NN_TENSOR_FLAG_INT8_CONVERSION; in TransformHardSwishIntoSupportedOps()
897 AddTensorInput(lite_input_index, false, tensor_flags)); in TransformHardSwishIntoSupportedOps()
921 AddTensorInput(lite_input_index, false, tensor_flags)); in TransformHardSwishIntoSupportedOps()
974 AddTensorOutput(lite_output_index, tensor_flags)); in TransformHardSwishIntoSupportedOps()
1258 std::vector<uint32_t>* indices, int tensor_flags = 0) { in AddTensor() argument
[all …]
|