Home
last modified time | relevance | path

Searched refs:unextended_input1_shape (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbatch_to_space_nd.h39 inline void BatchToSpaceND(const RuntimeShape& unextended_input1_shape, in BatchToSpaceND() argument
48 TFLITE_DCHECK_GE(unextended_input1_shape.DimensionsCount(), 3); in BatchToSpaceND()
49 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BatchToSpaceND()
50 TFLITE_DCHECK_EQ(unextended_input1_shape.DimensionsCount(), in BatchToSpaceND()
54 ExtendShapeBatchToSpace(unextended_input1_shape); in BatchToSpaceND()
69 unextended_input1_shape.DimensionsCount() == 4 ? block_shape_data[1] : 1; in BatchToSpaceND()
72 unextended_input1_shape.DimensionsCount() == 4 ? crops_data[2] : 0; in BatchToSpaceND()
Dspace_to_batch_nd.h41 const RuntimeShape& unextended_input1_shape, in SpaceToBatchND() argument
50 TFLITE_DCHECK_GE(unextended_input1_shape.DimensionsCount(), 3); in SpaceToBatchND()
51 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in SpaceToBatchND()
52 TFLITE_DCHECK_EQ(unextended_input1_shape.DimensionsCount(), in SpaceToBatchND()
57 ExtendShapeSpaceToBatch(unextended_input1_shape); in SpaceToBatchND()
72 unextended_input1_shape.DimensionsCount() == 4 ? block_shape_data[1] : 1; in SpaceToBatchND()
75 unextended_input1_shape.DimensionsCount() == 4 ? paddings_data[2] : 0; in SpaceToBatchND()
Dmaximum_minimum.h25 void MaximumMinimumBroadcastSlow(const RuntimeShape& unextended_input1_shape, in MaximumMinimumBroadcastSlow() argument
32 if (unextended_input1_shape == unextended_input2_shape) { in MaximumMinimumBroadcastSlow()
34 MatchingElementsSize(unextended_input1_shape, unextended_input2_shape, in MaximumMinimumBroadcastSlow()
40 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), N); in MaximumMinimumBroadcastSlow()
48 unextended_input1_shape, unextended_input2_shape, &desc1, &desc2); in MaximumMinimumBroadcastSlow()
Ddiv.h75 const RuntimeShape& unextended_input1_shape, in BroadcastDivSlow() argument
81 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), N); in BroadcastDivSlow()
88 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastDivSlow()
133 const RuntimeShape& unextended_input1_shape, in BroadcastDivSlow() argument
143 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), N); in BroadcastDivSlow()
150 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastDivSlow()
Dbinary_function.h31 const RuntimeShape& unextended_input1_shape, const T1* input1_data, in BroadcastBinaryFunction4DSlow() argument
35 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastBinaryFunction4DSlow()
43 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastBinaryFunction4DSlow()
Dcomparisons.h116 const RuntimeShape& unextended_input1_shape, in BroadcastComparison4DSlowPreprocess() argument
119 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastComparison4DSlowPreprocess()
124 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastComparison4DSlowPreprocess()
133 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BroadcastComparison4DSlowImpl() argument
137 BroadcastComparison4DSlowPreprocess(unextended_input1_shape, in BroadcastComparison4DSlowImpl()
170 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BroadcastComparison4DSlowWithScaling() argument
174 BroadcastComparison4DSlowPreprocess(unextended_input1_shape, in BroadcastComparison4DSlowWithScaling()
Dmul.h116 const RuntimeShape& unextended_input1_shape, in BroadcastMul4DSlow() argument
126 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastMul4DSlow()
134 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastMul4DSlow()
Dstring_comparisons.h58 const RuntimeShape& unextended_input1_shape, const TfLiteTensor* input1, in BroadcastComparison4DSlowStringImpl() argument
62 BroadcastComparison4DSlowPreprocess(unextended_input1_shape, in BroadcastComparison4DSlowStringImpl()
Dreference_ops.h1659 inline void BroadcastPow4DSlow(const RuntimeShape& unextended_input1_shape, in BroadcastPow4DSlow() argument
1665 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastPow4DSlow()
1673 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastPow4DSlow()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dprelu.cc58 const RuntimeShape& unextended_input1_shape, const float* input1_data, in BroadcastPrelu4DSlowFloat() argument
61 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BroadcastPrelu4DSlowFloat()
69 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastPrelu4DSlowFloat()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2620 const RuntimeShape& unextended_input1_shape, in BroadcastDivSlow() argument
2631 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), N); in BroadcastDivSlow()
2638 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastDivSlow()
2669 const RuntimeShape& unextended_input1_shape, in BroadcastDivSlow() argument
2675 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), N); in BroadcastDivSlow()
2682 NdArrayDescsForElementwiseBroadcast(unextended_input1_shape, in BroadcastDivSlow()
4987 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BatchToSpaceND() argument
4993 TFLITE_DCHECK_GE(unextended_input1_shape.DimensionsCount(), 3); in BatchToSpaceND()
4994 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BatchToSpaceND()
4995 TFLITE_DCHECK_EQ(unextended_input1_shape.DimensionsCount(), in BatchToSpaceND()
[all …]