Home
last modified time | relevance | path

Searched refs:broadcast_category (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dprocess_broadcast_shapes.h46 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
54 params->broadcast_category = BroadcastableOpCategory::kNonBroadcast; in ProcessBroadcastShapes()
62 params->broadcast_category = in ProcessBroadcastShapes()
66 params->broadcast_category = in ProcessBroadcastShapes()
72 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
77 if (params->broadcast_category != in ProcessBroadcastShapes()
79 params->broadcast_category != in ProcessBroadcastShapes()
84 params->broadcast_category = BroadcastableOpCategory::kNonBroadcast; in ProcessBroadcastShapes()
90 const bool swap_inputs = params->broadcast_category == in ProcessBroadcastShapes()
132 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
H A Dadd.h320 unswitched_params.broadcast_category == in BroadcastAddFivefold()
H A Dreference_ops.h198 unswitched_params.broadcast_category == in BroadcastMulFivefold()
H A Dlegacy_reference_ops.h1402 op_params.broadcast_category = in BroadcastAddFivefold()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dsub.h142 unswitched_params.broadcast_category == in BroadcastSubFiveFold()
225 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastSubDispatch()
H A Dmul.h252 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMulDispatch()
H A Dadd.h499 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastAddDispatch()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtypes.h638 BroadcastableOpCategory broadcast_category; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h158 unswitched_params.broadcast_category == in BinaryBroadcastFiveFold()
1847 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastAddDispatch()
2245 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMulDispatch()
7525 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMaximumDispatch()
7544 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastMinimumDispatch()
7704 if (params.broadcast_category == BroadcastableOpCategory::kGenericBroadcast) { in BroadcastPReluDispatch()
H A Dlegacy_optimized_ops.h3637 op_params.broadcast_category = in BroadcastAddFivefold()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/kernels/libtflite_kernel_utils/android_arm64_armv8-2a_cortex-a55_static_apex30_sabi/obj/external/tensorflow/tensorflow/lite/kernels/internal/
Dtranspose_utils.sdump28367 "field_name" : "broadcast_category",
Dquantization_util.sdump29012 "field_name" : "broadcast_category",
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/kernels/libtflite_kernel_utils/android_arm64_armv8-2a_cortex-a55_static_apex30_sabi/obj/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.sdump66113 "field_name" : "broadcast_category",
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/kernels/libtflite_kernel_utils/android_arm64_armv8-2a_cortex-a55_static_apex30_sabi/obj/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.sdump66411 "field_name" : "broadcast_category",
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/kernels/libtflite_kernel_utils/android_arm64_armv8-2a_cortex-a55_static_apex30_sabi/obj/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.sdump110493 "field_name" : "broadcast_category",