| /external/tensorflow/tensorflow/lite/kernels/ |
| D | comparisons.cc | 47 bool requires_broadcast = !HaveSameShapes(input1, input2); in ComparisonPrepare() local
50 if (requires_broadcast) { in ComparisonPrepare()
66 bool requires_broadcast) { \
89 if (requires_broadcast) { \
112 #define TF_LITE_COMPARISON(type, opname, requires_broadcast) \ argument
115 requires_broadcast \
130 bool requires_broadcast = !HaveSameShapes(input1, input2); in EqualEval() local
133 TF_LITE_COMPARISON(bool, Equal, requires_broadcast); in EqualEval()
136 TF_LITE_COMPARISON(float, Equal, requires_broadcast); in EqualEval()
139 TF_LITE_COMPARISON(int32_t, Equal, requires_broadcast); in EqualEval()
[all …]
|
| D | floor_mod.cc | 37 bool requires_broadcast; member
62 data->requires_broadcast = false; in Init()
91 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
94 if (data->requires_broadcast) { in Prepare()
105 TfLiteStatus EvalImpl(TfLiteContext* context, bool requires_broadcast, in EvalImpl() argument
120 if (requires_broadcast) { in EvalImpl()
144 return EvalImpl<int32_t>(context, data->requires_broadcast, input1, in Eval()
148 return EvalImpl<int64_t>(context, data->requires_broadcast, input1, in Eval()
152 return EvalImpl<float>(context, data->requires_broadcast, input1, input2, in Eval()
|
| D | pow.cc | 34 bool requires_broadcast; member
39 data->requires_broadcast = false; in Init()
66 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
69 if (data->requires_broadcast) { in Prepare()
81 TfLiteTensor* output, bool requires_broadcast) { in PowImpl() argument
82 if (requires_broadcast) { in PowImpl()
118 PowImpl<int32_t>(input1, input2, output, data->requires_broadcast); in Eval()
122 PowImpl<float>(input1, input2, output, data->requires_broadcast); in Eval()
|
| D | div.cc | 41 bool requires_broadcast; member
46 data->requires_broadcast = false; in Init()
67 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
70 if (data->requires_broadcast) { in Prepare()
97 if (data->requires_broadcast) { in EvalDiv()
103 if (data->requires_broadcast) { in EvalDiv()
111 if (data->requires_broadcast) { in EvalDiv()
117 if (data->requires_broadcast) { in EvalDiv()
|
| D | floor_div.cc | 34 bool requires_broadcast; member
45 data->requires_broadcast = false; in Init()
73 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
76 if (data->requires_broadcast) { in Prepare()
87 TfLiteStatus EvalImpl(TfLiteContext* context, bool requires_broadcast, in EvalImpl() argument
99 if (requires_broadcast) { in EvalImpl()
123 return EvalImpl<int32_t>(context, data->requires_broadcast, input1, in Eval()
|
| D | logical.cc | 34 bool requires_broadcast; member
39 data->requires_broadcast = false; in Init()
67 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
70 if (data->requires_broadcast) { in Prepare()
88 if (data->requires_broadcast) { in LogicalImpl()
|
| D | squared_difference.cc | 34 bool requires_broadcast; member
45 data->requires_broadcast = false; in Init()
66 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
69 if (data->requires_broadcast) { in Prepare()
83 if (data->requires_broadcast) { in EvalSquaredDifference()
|
| D | mul.cc | 42 bool requires_broadcast; member
55 data->requires_broadcast = false; in Init()
76 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
79 if (data->requires_broadcast) { in Prepare()
126 if (data->requires_broadcast) { in EvalMul()
132 if (data->requires_broadcast) { in EvalMul()
140 if (data->requires_broadcast) { in EvalMul()
146 if (data->requires_broadcast) { in EvalMul()
|
| D | add.cc | 42 bool requires_broadcast; member
64 data->requires_broadcast = false; in Init()
86 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
89 if (data->requires_broadcast) { in Prepare()
192 if (data->requires_broadcast) { in EvalAdd()
198 if (data->requires_broadcast) { in EvalAdd()
206 if (data->requires_broadcast) { in EvalAdd()
212 if (data->requires_broadcast) { in EvalAdd()
|
| D | sub.cc | 43 bool requires_broadcast; member
65 data->requires_broadcast = false; in Init()
204 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
207 if (data->requires_broadcast) { in Prepare()
242 if (data->requires_broadcast) { in EvalSub()
248 if (data->requires_broadcast) { in EvalSub()
256 if (data->requires_broadcast) { in EvalSub()
262 if (data->requires_broadcast) { in EvalSub()
|
| D | maximum_minimum.cc | 57 bool requires_broadcast = in Prepare() local
61 if (requires_broadcast) { in Prepare()
|