Home
last modified time | relevance | path

Searched refs:ExtendedShape (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h76 RuntimeShape::ExtendedShape(4, unextended_input_shape); in DepthToSpace()
78 RuntimeShape::ExtendedShape(4, unextended_output_shape); in DepthToSpace()
129 RuntimeShape::ExtendedShape(4, unextended_input_shape); in SpaceToDepth()
131 RuntimeShape::ExtendedShape(4, unextended_output_shape); in SpaceToDepth()
494 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastDiv4DSlow()
604 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastDiv4DSlow()
706 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastSub4DSlow()
747 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastSub4DSlow()
811 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastSub4DSlow()
851 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastSub4DSlow()
[all …]
Dstrided_slice.h37 RuntimeShape::ExtendedShape(4, unextended_input_shape); in StridedSlice()
39 RuntimeShape::ExtendedShape(4, unextended_output_shape); in StridedSlice()
Dprocess_broadcast_shapes.h47 auto extended_shape0 = RuntimeShape::ExtendedShape(dims_count, shape0); in ProcessBroadcastShapes()
48 auto extended_shape1 = RuntimeShape::ExtendedShape(dims_count, shape1); in ProcessBroadcastShapes()
Dreduce.h217 RuntimeShape::ExtendedShape(4, unextended_input_shape); in Mean()
219 RuntimeShape::ExtendedShape(4, unextended_output_shape); in Mean()
262 RuntimeShape::ExtendedShape(4, unextended_input_shape); in Mean()
264 RuntimeShape::ExtendedShape(4, unextended_output_shape); in Mean()
Dpad.h43 RuntimeShape::ExtendedShape(PadKernelMaxDimensionCount(), input_shape); in PadImpl()
45 RuntimeShape::ExtendedShape(PadKernelMaxDimensionCount(), output_shape); in PadImpl()
Dmul.h86 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastMul4DSlow()
130 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastMul4DSlow()
Dmaximum_minimum.h35 RuntimeShape::ExtendedShape(4, unextended_output_shape); in MaximumMinimumBroadcast4DSlow()
Dadd.h191 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
231 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
272 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
Dprelu.h38 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastPrelu4DSlow()
Dbinary_function.h42 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastBinaryFunction4DSlow()
Dcomparisons.h118 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastComparison4DSlowImpl()
160 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastComparison4DSlowWithScaling()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmean.h33 RuntimeShape::ExtendedShape(4, unextended_input_shape); in Mean()
35 RuntimeShape::ExtendedShape(4, unextended_output_shape); in Mean()
Dadd.h89 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
Dmul.h100 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastMul4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h603 auto extended_input0_shape = RuntimeShape::ExtendedShape(N, input0_shape); in NdArrayDescsForElementwiseBroadcast()
604 auto extended_input1_shape = RuntimeShape::ExtendedShape(N, input1_shape); in NdArrayDescsForElementwiseBroadcast()
638 auto extended_input0_shape = RuntimeShape::ExtendedShape(N, input0_shape); in NdArrayDescsForElementwiseBroadcast()
639 auto extended_input1_shape = RuntimeShape::ExtendedShape(N, input1_shape); in NdArrayDescsForElementwiseBroadcast()
640 auto extended_input2_shape = RuntimeShape::ExtendedShape(N, input2_shape); in NdArrayDescsForElementwiseBroadcast()
Dtypes.h254 inline static RuntimeShape ExtendedShape(int new_shape_size, in ExtendedShape() function
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmean.h190 RuntimeShape::ExtendedShape(4, unextended_input_shape); in Mean()
192 RuntimeShape::ExtendedShape(4, unextended_output_shape); in Mean()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dprelu.cc40 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastPrelu4DSlowFloat()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1090 RuntimeShape::ExtendedShape(4, unextended_input_shape); in Mean()
1092 RuntimeShape::ExtendedShape(4, unextended_output_shape); in Mean()
1518 RuntimeShape::ExtendedShape(4, unextended_input_shape); in DepthToSpace()
1520 RuntimeShape::ExtendedShape(4, unextended_output_shape); in DepthToSpace()
1559 RuntimeShape::ExtendedShape(4, unextended_input_shape); in SpaceToDepth()
1561 RuntimeShape::ExtendedShape(4, unextended_output_shape); in SpaceToDepth()
2759 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastDiv4DSlow()
2806 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastDiv4DSlow()
2953 RuntimeShape::ExtendedShape(4, unextended_input_shape); in LstmCell()
2955 RuntimeShape::ExtendedShape(4, unextended_prev_activ_shape); in LstmCell()
[all …]
Dlegacy_optimized_ops.h2987 RuntimeShape::ExtendedShape(4, unextended_input_shape); in LstmCell()
2989 RuntimeShape::ExtendedShape(4, unextended_prev_activ_shape); in LstmCell()
2991 RuntimeShape::ExtendedShape(4, unextended_bias_shape); in LstmCell()
2993 RuntimeShape::ExtendedShape(4, unextended_prev_state_shape); in LstmCell()
2995 RuntimeShape::ExtendedShape(4, unextended_output_state_shape); in LstmCell()
2997 RuntimeShape::ExtendedShape(4, unextended_output_activ_shape); in LstmCell()
2999 RuntimeShape::ExtendedShape(4, unextended_concat_temp_shape); in LstmCell()
3001 RuntimeShape::ExtendedShape(4, unextended_activ_temp_shape); in LstmCell()
3138 RuntimeShape::ExtendedShape(4, unextended_input_shape); in LstmCell()
3140 RuntimeShape::ExtendedShape(4, unextended_prev_activ_shape); in LstmCell()
[all …]