Home
last modified time | relevance | path

Searched refs:BroadcastableOpCategory (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dprocess_broadcast_shapes.h44 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
52 params->broadcast_category = BroadcastableOpCategory::kNonBroadcast; in ProcessBroadcastShapes()
61 BroadcastableOpCategory::kFirstInputBroadcastsFast; in ProcessBroadcastShapes()
65 BroadcastableOpCategory::kSecondInputBroadcastsFast; in ProcessBroadcastShapes()
70 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
76 BroadcastableOpCategory::kFirstInputBroadcastsFast && in ProcessBroadcastShapes()
78 BroadcastableOpCategory::kSecondInputBroadcastsFast) { in ProcessBroadcastShapes()
85 BroadcastableOpCategory::kSecondInputBroadcastsFast; in ProcessBroadcastShapes()
126 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
Dadd.h340 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
Dlegacy_reference_ops.h1430 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
Dreference_ops.h387 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastMulFivefold()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h199 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastMulFivefold()
262 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMulDispatch()
Dadd.h253 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
336 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastAddDispatch()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h716 enum class BroadcastableOpCategory : uint8 { enum
755 BroadcastableOpCategory broadcast_category;
/external/tensorflow/tensorflow/lite/kernels/
Dsub.cc313 BroadcastableOpCategory::kGenericBroadcast) { in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2052 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
2139 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
2218 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastAddDispatch()
2597 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastMulFivefold()
2664 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastMulFivefold()
2725 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMulDispatch()
Dlegacy_optimized_ops.h3624 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()