Home
last modified time | relevance | path

Searched refs:SubscriptToIndex (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsub.h99 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastSubSlow()
101 input1_data[SubscriptToIndex(desc1, indexes)] - in BroadcastSubSlow()
102 input2_data[SubscriptToIndex(desc2, indexes)], in BroadcastSubSlow()
140 params.input1_offset + input1_data[SubscriptToIndex(desc1, indexes)]; in BroadcastSubSlow()
142 params.input2_offset + input2_data[SubscriptToIndex(desc2, indexes)]; in BroadcastSubSlow()
159 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastSubSlow()
196 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastSubSlow()
198 input1_data[SubscriptToIndex(desc1, indexes)] - in BroadcastSubSlow()
199 input2_data[SubscriptToIndex(desc2, indexes)], in BroadcastSubSlow()
234 params.input1_offset + input1_data[SubscriptToIndex(desc1, indexes)]; in BroadcastSubSlow()
[all …]
Ddiv.h102 params.input1_offset + input1_data[SubscriptToIndex(desc1, indexes)]; in BroadcastDivSlow()
104 params.input2_offset + input2_data[SubscriptToIndex(desc2, indexes)]; in BroadcastDivSlow()
121 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastDivSlow()
164 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastDivSlow()
166 input1_data[SubscriptToIndex(desc1, indexes)] / in BroadcastDivSlow()
167 input2_data[SubscriptToIndex(desc2, indexes)], in BroadcastDivSlow()
Dmaximum_minimum.h53 output_data[SubscriptToIndex(output_desc, indexes)] = in MaximumMinimumBroadcastSlow()
54 op(input1_data[SubscriptToIndex(desc1, indexes)], in MaximumMinimumBroadcastSlow()
55 input2_data[SubscriptToIndex(desc2, indexes)]); in MaximumMinimumBroadcastSlow()
Dbroadcast_to.h32 input_data + SubscriptToIndex(input_desc, indexes) * type_size; in BroadcastImpl()
34 output_data + SubscriptToIndex(output_desc, indexes) * type_size; in BroadcastImpl()
53 output_data + SubscriptToIndex(output_desc, indexes) * type_size; in BroadcastImpl()
Dmul.h94 input1_data[SubscriptToIndex(desc1, b, y, x, c)]; in BroadcastMul4DSlow()
97 input2_data[SubscriptToIndex(desc2, b, y, x, c)]; in BroadcastMul4DSlow()
154 input1_data[SubscriptToIndex(desc1, b, y, x, c)] * in BroadcastMul4DSlow()
155 input2_data[SubscriptToIndex(desc2, b, y, x, c)], in BroadcastMul4DSlow()
Dbinary_function.h51 auto in1_idx = SubscriptToIndex(desc1, b, y, x, c); in BroadcastBinaryFunction4DSlow()
52 auto in2_idx = SubscriptToIndex(desc2, b, y, x, c); in BroadcastBinaryFunction4DSlow()
Dstring_comparisons.h71 GetString(input1, SubscriptToIndex(dims.desc1, b, y, x, c)); in BroadcastComparison4DSlowStringImpl()
73 GetString(input2, SubscriptToIndex(dims.desc2, b, y, x, c)); in BroadcastComparison4DSlowStringImpl()
Dadd.h236 input1_data[SubscriptToIndex(desc1, b, y, x, c)] + in BroadcastAdd4DSlow()
237 input2_data[SubscriptToIndex(desc2, b, y, x, c)], in BroadcastAdd4DSlow()
276 input1_data[SubscriptToIndex(desc1, b, y, x, c)] + in BroadcastAdd4DSlow()
277 input2_data[SubscriptToIndex(desc2, b, y, x, c)], in BroadcastAdd4DSlow()
318 input1_data[SubscriptToIndex(desc1, b, y, x, c)]; in BroadcastAdd4DSlow()
321 input2_data[SubscriptToIndex(desc2, b, y, x, c)]; in BroadcastAdd4DSlow()
Dtranspose.h66 output_data[SubscriptToIndex(output_desc, indexes)] = in TransposeImpl()
67 input_data[SubscriptToIndex(perm_input_desc, indexes)]; in TransposeImpl()
Dcomparisons.h146 F(input1_data[SubscriptToIndex(dims.desc1, b, y, x, c)], in BroadcastComparison4DSlowImpl()
147 input2_data[SubscriptToIndex(dims.desc2, b, y, x, c)]); in BroadcastComparison4DSlowImpl()
192 input1_data[SubscriptToIndex(dims.desc1, b, y, x, c)]; in BroadcastComparison4DSlowWithScaling()
195 input2_data[SubscriptToIndex(dims.desc2, b, y, x, c)]; in BroadcastComparison4DSlowWithScaling()
Dprelu.h46 int input_index = SubscriptToIndex(desc1, b, y, x, c); in BroadcastPrelu4DSlow()
54 auto alpha_index = SubscriptToIndex(desc2, b, y, x, c); in BroadcastPrelu4DSlow()
Dreference_ops.h1562 SubscriptToIndex(desc_condition, b, y, x, c); in BroadcastSelect4DSlow()
1563 const int x_index = SubscriptToIndex(desc_x, b, y, x, c); in BroadcastSelect4DSlow()
1564 const int y_index = SubscriptToIndex(desc_y, b, y, x, c); in BroadcastSelect4DSlow()
1681 auto in1_idx = SubscriptToIndex(desc1, b, y, x, c); in BroadcastPow4DSlow()
1682 auto in2_idx = SubscriptToIndex(desc2, b, y, x, c); in BroadcastPow4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h109 input1_data[SubscriptToIndex(desc1, b, y, x, c)]; in BroadcastMul4DSlow()
112 input2_data[SubscriptToIndex(desc2, b, y, x, c)]; in BroadcastMul4DSlow()
Dadd.h78 input1_data[SubscriptToIndex(desc1, b, y, x, c)], in BroadcastBinaryFunction4DSlow()
79 input2_data[SubscriptToIndex(desc2, b, y, x, c)], params); in BroadcastBinaryFunction4DSlow()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dprelu.cc77 auto in1_idx = SubscriptToIndex(desc1, b, y, x, c); in BroadcastPrelu4DSlowFloat()
78 auto in2_idx = SubscriptToIndex(desc2, b, y, x, c); in BroadcastPrelu4DSlowFloat()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h725 inline int SubscriptToIndex(const NdArrayDesc<4>& desc, int i0, int i1, int i2, in SubscriptToIndex() function
735 inline int SubscriptToIndex(const NdArrayDesc<5>& desc, int indexes[5]) { in SubscriptToIndex() function
741 inline int SubscriptToIndex(const NdArrayDesc<8>& desc, int indexes[8]) { in SubscriptToIndex() function
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2655 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastDivSlow()
2657 input1_data[SubscriptToIndex(desc1, indexes)] / in BroadcastDivSlow()
2658 input2_data[SubscriptToIndex(desc2, indexes)], in BroadcastDivSlow()
2696 params.input1_offset + input1_data[SubscriptToIndex(desc1, indexes)]; in BroadcastDivSlow()
2698 params.input2_offset + input2_data[SubscriptToIndex(desc2, indexes)]; in BroadcastDivSlow()
2715 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastDivSlow()