Searched refs:unextended_input1_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 961 const RuntimeShape& unextended_input1_shape, in BroadcastMul4DSlow() argument
972 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastMul4DSlow()
980 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastMul4DSlow()
1191 const RuntimeShape& unextended_input1_shape, in BroadcastDiv4DSlow() argument
1201 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastDiv4DSlow()
1209 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastDiv4DSlow()
3077 const RuntimeShape& unextended_input1_shape, const T* input1_data, in SpaceToBatchND() argument
3082 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in SpaceToBatchND()
3085 RuntimeShape::ExtendedShape(4, unextended_input1_shape); in SpaceToBatchND()
3135 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BatchToSpaceND() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3429 const RuntimeShape& unextended_input1_shape, in BroadcastDiv4DSlow() argument
3440 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastDiv4DSlow()
3448 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastDiv4DSlow()
5780 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BatchToSpaceND() argument
5786 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BatchToSpaceND()
5789 RuntimeShape::ExtendedShape(4, unextended_input1_shape); in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3505 const RuntimeShape& unextended_input1_shape, in BroadcastDiv4DSlow() argument
3516 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastDiv4DSlow()
3524 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastDiv4DSlow()
6108 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BatchToSpaceND() argument
6114 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BatchToSpaceND()
6117 RuntimeShape::ExtendedShape(4, unextended_input1_shape); in BatchToSpaceND()
|