Home
last modified time | relevance | path

Searched refs:kGenericBroadcast (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dprocess_broadcast_shapes.h44 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
70 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
126 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h262 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMulDispatch()
Dadd.h336 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastAddDispatch()
/external/tensorflow/tensorflow/lite/kernels/
Dsub.cc313 BroadcastableOpCategory::kGenericBroadcast) { in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h721 kGenericBroadcast, // Fall-back. enumerator
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2218 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastAddDispatch()
2725 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMulDispatch()