Searched defs: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
303 TfLiteIntArray* weights_dims = IntArrayFromInts(weights_dims_data); in TestFullyConnectedQuantized() local
|
| /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
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 409 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
427 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
522 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
572 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
596 const Dims<4>& weights_dims, const int32* bias_data_int32, 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()
704 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
2340 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
3099 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
3410 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell()
|