Home
last modified time | relevance | path

Searched refs:broadcast_category (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()
60 params->broadcast_category = in ProcessBroadcastShapes()
64 params->broadcast_category = in ProcessBroadcastShapes()
70 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
75 if (params->broadcast_category != in ProcessBroadcastShapes()
77 params->broadcast_category != in ProcessBroadcastShapes()
82 params->broadcast_category = BroadcastableOpCategory::kNonBroadcast; in ProcessBroadcastShapes()
88 const bool swap_inputs = params->broadcast_category == in ProcessBroadcastShapes()
130 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
Dadd.h366 unswitched_params.broadcast_category == in BroadcastAddFivefold()
Dreference_ops.h180 unswitched_params.broadcast_category == in BroadcastMulFivefold()
Dlegacy_reference_ops.h1399 op_params.broadcast_category = in BroadcastAddFivefold()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h252 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMulDispatch()
Dadd.h291 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastAddDispatch()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h793 BroadcastableOpCategory broadcast_category; member
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h54 uint8_t broadcast_category; // BroadcastableOpCategory broadcast_category; member
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h222 unswitched_params.broadcast_category == in BinaryBroadcastFiveFold()
2174 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastAddDispatch()
2572 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMulDispatch()
8013 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMaximumDispatch()
8032 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMinimumDispatch()
8192 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastPReluDispatch()
Dlegacy_optimized_ops.h3636 op_params.broadcast_category = in BroadcastAddFivefold()