Searched refs:unextended_concat_temp_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1844 const RuntimeShape& unextended_concat_temp_shape, float* concat_temp_data, in LstmCell() argument
1852 TFLITE_DCHECK_LE(unextended_concat_temp_shape.DimensionsCount(), 4); in LstmCell()
1867 RuntimeShape::ExtendedShape(4, unextended_concat_temp_shape); in LstmCell()
2050 const RuntimeShape& unextended_concat_temp_shape, in LstmCell() argument
2064 TFLITE_DCHECK_LE(unextended_concat_temp_shape.DimensionsCount(), 4); in LstmCell()
2079 RuntimeShape::ExtendedShape(4, unextended_concat_temp_shape); in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3653 const RuntimeShape& unextended_concat_temp_shape, float* concat_temp_data, in LstmCell() argument
3662 TFLITE_DCHECK_LE(unextended_concat_temp_shape.DimensionsCount(), 4); in LstmCell()
3677 RuntimeShape::ExtendedShape(4, unextended_concat_temp_shape); in LstmCell()
3803 const RuntimeShape& unextended_concat_temp_shape, in LstmCell() argument
3818 TFLITE_DCHECK_LE(unextended_concat_temp_shape.DimensionsCount(), 4); in LstmCell()
3833 RuntimeShape::ExtendedShape(4, unextended_concat_temp_shape); in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3562 const RuntimeShape& unextended_concat_temp_shape, float* concat_temp_data, in LstmCell() argument
3571 TFLITE_DCHECK_LE(unextended_concat_temp_shape.DimensionsCount(), 4); in LstmCell()
3586 RuntimeShape::ExtendedShape(4, unextended_concat_temp_shape); in LstmCell()
3687 const RuntimeShape& unextended_concat_temp_shape, in LstmCell() argument
3702 TFLITE_DCHECK_LE(unextended_concat_temp_shape.DimensionsCount(), 4); in LstmCell()
3717 RuntimeShape::ExtendedShape(4, unextended_concat_temp_shape); in LstmCell()
|