Searched defs:weights_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 411 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
429 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
524 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
574 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
598 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 687 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
705 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
2341 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
3100 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
3411 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell()
|