Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h675 enum class BroadcastableOpCategory : uint8 { enum
699 BroadcastableOpCategory broadcast_category;
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h691 enum class BroadcastableOpCategory : uint8 { enum
723 BroadcastableOpCategory broadcast_category;
/external/tensorflow/tensorflow/lite/kernels/
Dadd.cc264 BroadcastableOpCategory::kGenericBroadcast) { in EvalAddQuantized()
Dsub.cc316 BroadcastableOpCategory::kGenericBroadcast) { in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h48 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
56 params->broadcast_category = BroadcastableOpCategory::kNonBroadcast; in ProcessBroadcastShapes()
65 BroadcastableOpCategory::kFirstInputBroadcastsFast; in ProcessBroadcastShapes()
69 BroadcastableOpCategory::kSecondInputBroadcastsFast; in ProcessBroadcastShapes()
72 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
78 BroadcastableOpCategory::kFirstInputBroadcastsFast && in ProcessBroadcastShapes()
80 BroadcastableOpCategory::kSecondInputBroadcastsFast) { in ProcessBroadcastShapes()
87 BroadcastableOpCategory::kSecondInputBroadcastsFast; in ProcessBroadcastShapes()
128 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
857 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
[all …]
Dlegacy_reference_ops.h1309 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1074 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
Doptimized_ops.h2969 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
3363 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastMulFivefold()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h3082 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastAddFivefold()
3439 tflite::BroadcastableOpCategory::kFirstInputBroadcastsFast; in BroadcastMulFivefold()