Searched refs:input_weight_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 3243 TfLiteIntArray* input_weight_dims = TfLiteIntArrayCreate(2); in Map() local
3245 weight_tensor.dims, recurrent_weight_dims, input_weight_dims); in Map()
3251 input_weight_dims, input_to_input, weight_tensor.params, in Map()
3256 input_weight_dims, input_to_forget, weight_tensor.params, in Map()
3261 input_weight_dims, input_to_cell, weight_tensor.params, in Map()
3266 input_weight_dims, input_to_output, weight_tensor.params, in Map()
3289 TfLiteIntArrayFree(input_weight_dims); in Map()
|