Home
last modified time | relevance | path

Searched refs:CopyDimsToDesc (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsub.h85 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
125 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
164 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
201 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
238 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSub16POTSlow()
284 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastQuantSubSlow()
Dbroadcast_to.h68 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_input_shape), in BroadcastTo()
70 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastTo()
Dtranspose.h40 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_input_shape), in TransposeImpl()
42 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in TransposeImpl()
Dmaximum_minimum.h49 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in MaximumMinimumBroadcastSlow()
Ddiv.h117 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlowQuantized()
205 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlow()
Dselect.h97 CopyDimsToDesc(extended_output_shape, &desc_output); in BroadcastSelect5DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h950 inline void CopyDimsToDesc(const RuntimeShape& input_shape, in CopyDimsToDesc() function
971 CopyDimsToDesc<N>(extended_input0_shape, desc0_out); in NdArrayDescsForElementwiseBroadcast()
972 CopyDimsToDesc<N>(extended_input1_shape, desc1_out); in NdArrayDescsForElementwiseBroadcast()
1007 CopyDimsToDesc<N>(extended_input0_shape, desc0_out); in NdArrayDescsForElementwiseBroadcast()
1008 CopyDimsToDesc<N>(extended_input1_shape, desc1_out); in NdArrayDescsForElementwiseBroadcast()
1009 CopyDimsToDesc<N>(extended_input2_shape, desc2_out); in NdArrayDescsForElementwiseBroadcast()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2313 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlow()
2357 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlow()