Home
last modified time | relevance | path

Searched refs:NdArrayDesc (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsub.h80 NdArrayDesc<N> desc1; in BroadcastSubSlow()
81 NdArrayDesc<N> desc2; in BroadcastSubSlow()
82 NdArrayDesc<N> output_desc; in BroadcastSubSlow()
120 NdArrayDesc<N> desc1; in BroadcastSubSlow()
121 NdArrayDesc<N> desc2; in BroadcastSubSlow()
122 NdArrayDesc<N> output_desc; in BroadcastSubSlow()
177 NdArrayDesc<N> desc1; in BroadcastSubSlow()
178 NdArrayDesc<N> desc2; in BroadcastSubSlow()
179 NdArrayDesc<N> output_desc; in BroadcastSubSlow()
214 NdArrayDesc<N> desc1; in BroadcastSubSlow()
[all …]
Dbroadcast_to.h24 void BroadcastImpl(const NdArrayDesc<N>& input_desc, const char* input_data, in BroadcastImpl()
25 const NdArrayDesc<N>& output_desc, char* output_data, in BroadcastImpl()
66 NdArrayDesc<N> input_desc; in BroadcastTo()
67 NdArrayDesc<N> output_desc; in BroadcastTo()
Ddiv.h85 NdArrayDesc<N> desc1; in BroadcastDivSlow()
86 NdArrayDesc<N> desc2; in BroadcastDivSlow()
87 NdArrayDesc<N> output_desc; in BroadcastDivSlow()
147 NdArrayDesc<N> desc1; in BroadcastDivSlow()
148 NdArrayDesc<N> desc2; in BroadcastDivSlow()
149 NdArrayDesc<N> output_desc; in BroadcastDivSlow()
Dmaximum_minimum.h44 NdArrayDesc<N> desc1; in MaximumMinimumBroadcastSlow()
45 NdArrayDesc<N> desc2; in MaximumMinimumBroadcastSlow()
46 NdArrayDesc<N> output_desc; in MaximumMinimumBroadcastSlow()
Dmul.h81 NdArrayDesc<4> desc1; in BroadcastMul4DSlow()
82 NdArrayDesc<4> desc2; in BroadcastMul4DSlow()
132 NdArrayDesc<4> desc1; in BroadcastMul4DSlow()
133 NdArrayDesc<4> desc2; in BroadcastMul4DSlow()
Dtranspose.h38 NdArrayDesc<N> input_desc; in TransposeImpl()
39 NdArrayDesc<N> output_desc; in TransposeImpl()
56 NdArrayDesc<N> perm_input_desc; in TransposeImpl()
Dbinary_function.h41 NdArrayDesc<4> desc1; in BroadcastBinaryFunction4DSlow()
42 NdArrayDesc<4> desc2; in BroadcastBinaryFunction4DSlow()
Dadd.h212 NdArrayDesc<4> desc1; in BroadcastAdd4DSlow()
213 NdArrayDesc<4> desc2; in BroadcastAdd4DSlow()
252 NdArrayDesc<4> desc1; in BroadcastAdd4DSlow()
253 NdArrayDesc<4> desc2; in BroadcastAdd4DSlow()
294 NdArrayDesc<4> desc1; in BroadcastAdd4DSlow()
295 NdArrayDesc<4> desc2; in BroadcastAdd4DSlow()
Dcomparisons.h111 NdArrayDesc<4> desc1;
112 NdArrayDesc<4> desc2;
122 NdArrayDesc<4> desc1; in BroadcastComparison4DSlowPreprocess()
123 NdArrayDesc<4> desc2; in BroadcastComparison4DSlowPreprocess()
Dprelu.h37 NdArrayDesc<4> desc1; in BroadcastPrelu4DSlow()
38 NdArrayDesc<4> desc2; in BroadcastPrelu4DSlow()
Dreference_ops.h1539 NdArrayDesc<4> desc_condition; in BroadcastSelect4DSlow()
1540 NdArrayDesc<4> desc_x; in BroadcastSelect4DSlow()
1541 NdArrayDesc<4> desc_y; in BroadcastSelect4DSlow()
1671 NdArrayDesc<4> desc1; in BroadcastPow4DSlow()
1672 NdArrayDesc<4> desc2; in BroadcastPow4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h711 struct NdArrayDesc { struct
725 inline int SubscriptToIndex(const NdArrayDesc<4>& desc, int i0, int i1, int i2, in SubscriptToIndex()
735 inline int SubscriptToIndex(const NdArrayDesc<5>& desc, int indexes[5]) { in SubscriptToIndex()
741 inline int SubscriptToIndex(const NdArrayDesc<8>& desc, int indexes[8]) { in SubscriptToIndex()
773 NdArrayDesc<N>* desc0_out, in NdArrayDescsForElementwiseBroadcast()
774 NdArrayDesc<N>* desc1_out) { in NdArrayDescsForElementwiseBroadcast()
808 NdArrayDesc<N>* desc_out) { in CopyDimsToDesc()
820 NdArrayDesc<N>* desc0_out, NdArrayDesc<N>* desc1_out) { in NdArrayDescsForElementwiseBroadcast()
853 const RuntimeShape& input2_shape, NdArrayDesc<N>* desc0_out, in NdArrayDescsForElementwiseBroadcast()
854 NdArrayDesc<N>* desc1_out, NdArrayDesc<N>* desc2_out) { in NdArrayDescsForElementwiseBroadcast()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h95 NdArrayDesc<4> desc1; in BroadcastMul4DSlow()
96 NdArrayDesc<4> desc2; in BroadcastMul4DSlow()
Dadd.h55 NdArrayDesc<4> desc1; in BroadcastBinaryFunction4DSlow()
56 NdArrayDesc<4> desc2; in BroadcastBinaryFunction4DSlow()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dprelu.cc67 NdArrayDesc<4> desc1; in BroadcastPrelu4DSlowFloat()
68 NdArrayDesc<4> desc2; in BroadcastPrelu4DSlowFloat()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2635 NdArrayDesc<N> desc1; in BroadcastDivSlow()
2636 NdArrayDesc<N> desc2; in BroadcastDivSlow()
2637 NdArrayDesc<N> output_desc; in BroadcastDivSlow()
2679 NdArrayDesc<N> desc1; in BroadcastDivSlow()
2680 NdArrayDesc<N> desc2; in BroadcastDivSlow()
2681 NdArrayDesc<N> output_desc; in BroadcastDivSlow()