Searched refs:weights_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | fully_connected_test.cc | 271 TfLiteIntArray* weights_dims = IntArrayFromInts(weights_dims_data); in TestFullyConnectedFloat() local
281 CreateTensor(weights_data, weights_dims), in TestFullyConnectedFloat()
303 TfLiteIntArray* weights_dims = IntArrayFromInts(weights_dims_data); in TestFullyConnectedQuantized() local
314 CreateQuantizedTensor(weights_data, weights_quantized, weights_dims, in TestFullyConnectedQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/ |
| D | fully_connected_slicing_test.cc | 47 TfLiteIntArray* weights_dims = IntArrayFromInts(weights_dims_data); in TestFullyConnectedQuantized() local
58 CreateQuantizedTensor(weights_data, weights_dims, "weights_tensor", in TestFullyConnectedQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 409 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected() argument
419 DimsToShape(weights_dims), weights_data, in FullyConnected()
427 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
432 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
522 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument
536 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()
572 const Dims<4>& weights_dims, const float* bias_data, in LstmCell() argument
584 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims), in LstmCell()
596 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell() argument
613 DimsToShape(weights_dims), weights_data_uint8, DimsToShape(bias_dims), in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 686 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected() argument
696 DimsToShape(weights_dims), weights_data, in FullyConnected()
704 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
709 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
2340 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument
2354 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()
3099 const Dims<4>& weights_dims, const float* bias_data, in LstmCell() argument
3111 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims), in LstmCell()
3410 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell() argument
3427 DimsToShape(weights_dims), weights_data_uint8, DimsToShape(bias_dims), in LstmCell()
|