Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbroadcast_to.h68 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_input_shape), in BroadcastTo()
70 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastTo()
Dsub.h85 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
125 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
182 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
219 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
275 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
312 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
349 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSub16POTSlow()
Dtranspose.h40 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_input_shape), in TransposeImpl()
42 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in TransposeImpl()
Ddiv.h90 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlow()
152 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlow()
Dmaximum_minimum.h49 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in MaximumMinimumBroadcastSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h807 inline void CopyDimsToDesc(const RuntimeShape& input_shape, in CopyDimsToDesc() function
828 CopyDimsToDesc<N>(extended_input0_shape, desc0_out); in NdArrayDescsForElementwiseBroadcast()
829 CopyDimsToDesc<N>(extended_input1_shape, desc1_out); in NdArrayDescsForElementwiseBroadcast()
864 CopyDimsToDesc<N>(extended_input0_shape, desc0_out); in NdArrayDescsForElementwiseBroadcast()
865 CopyDimsToDesc<N>(extended_input1_shape, desc1_out); in NdArrayDescsForElementwiseBroadcast()
866 CopyDimsToDesc<N>(extended_input2_shape, desc2_out); in NdArrayDescsForElementwiseBroadcast()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2640 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlow()
2684 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlow()