Searched refs:unextended_bias_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1839 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell() argument
1848 TFLITE_DCHECK_LE(unextended_bias_shape.DimensionsCount(), 4); in LstmCell()
1859 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
2042 const uint8* weights_data_uint8, const RuntimeShape& unextended_bias_shape, in LstmCell() argument
2060 TFLITE_DCHECK_LE(unextended_bias_shape.DimensionsCount(), 4); in LstmCell()
2071 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3557 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell() argument
3567 TFLITE_DCHECK_LE(unextended_bias_shape.DimensionsCount(), 4); in LstmCell()
3578 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
3679 const uint8* weights_data_uint8, const RuntimeShape& unextended_bias_shape, in LstmCell() argument
3698 TFLITE_DCHECK_LE(unextended_bias_shape.DimensionsCount(), 4); in LstmCell()
3709 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3648 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell() argument
3658 TFLITE_DCHECK_LE(unextended_bias_shape.DimensionsCount(), 4); in LstmCell()
3669 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
3795 const uint8* weights_data_uint8, const RuntimeShape& unextended_bias_shape, in LstmCell() argument
3814 TFLITE_DCHECK_LE(unextended_bias_shape.DimensionsCount(), 4); in LstmCell()
3825 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
|