Searched defs:requires_broadcast (Results 1 – 14 of 14) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | comparisons.cc | 62 bool requires_broadcast = !HaveSameShapes(input1, input2); in ComparisonPrepareCommon() local 97 TfLiteTensor* output, bool requires_broadcast) { in ComparisonQuantized() 134 TfLiteTensor* output, bool requires_broadcast) { in Comparison() 149 TfLiteTensor* output, bool requires_broadcast) { in ComparisonString() 172 bool requires_broadcast = !HaveSameShapes(input1, input2); in EqualEval() local 222 bool requires_broadcast = !HaveSameShapes(input1, input2); in NotEqualEval() local 272 bool requires_broadcast = !HaveSameShapes(input1, input2); in GreaterEval() local 313 bool requires_broadcast = !HaveSameShapes(input1, input2); in GreaterEqualEval() local 354 bool requires_broadcast = !HaveSameShapes(input1, input2); in LessEval() local 395 bool requires_broadcast = !HaveSameShapes(input1, input2); in LessEqualEval() local
|
H A D | sub.cc | 57 bool requires_broadcast; member 304 bool requires_broadcast, TfLiteTensor* output) { in EvalSubImpl() 346 const bool requires_broadcast = data->requires_broadcast; in EvalSub() local
|
H A D | floor_div.cc | 41 bool requires_broadcast; member 99 TfLiteStatus EvalImpl(TfLiteContext* context, bool requires_broadcast, in EvalImpl()
|
H A D | floor_mod.cc | 39 bool requires_broadcast; member 95 TfLiteStatus EvalImpl(TfLiteContext* context, bool requires_broadcast, in EvalImpl()
|
H A D | pow.cc | 38 bool requires_broadcast; member 92 TfLiteTensor* output, bool requires_broadcast) { in PowImpl()
|
H A D | logical.cc | 37 bool requires_broadcast; member
|
H A D | select.cc | 40 bool requires_broadcast; member
|
H A D | maximum_minimum.cc | 64 bool requires_broadcast = in Prepare() local
|
H A D | squared_difference.cc | 41 bool requires_broadcast; member
|
H A D | div.cc | 49 bool requires_broadcast; member
|
H A D | mul.cc | 92 const bool requires_broadcast = !HaveSameShapes(input1, input2); in Prepare() local
|
H A D | add.cc | 107 const bool requires_broadcast = !HaveSameShapes(input1, input2); in Prepare() local
|
H A D | activations.cc | 108 bool requires_broadcast; member
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | xla_argument.h | 127 bool requires_broadcast = false; member
|