Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dadd.cc263 if (op_params.broadcast_category == in EvalAddQuantized()
Dsub.cc315 if (op_params.broadcast_category == 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()
64 params->broadcast_category = in ProcessBroadcastShapes()
68 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()
86 const bool swap_inputs = params->broadcast_category == in ProcessBroadcastShapes()
128 params->broadcast_category = BroadcastableOpCategory::kGenericBroadcast; in ProcessBroadcastShapes()
856 unswitched_params.broadcast_category == in BroadcastAddFivefold()
[all …]
Dlegacy_reference_ops.h1308 op_params.broadcast_category = in BroadcastAddFivefold()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h699 BroadcastableOpCategory broadcast_category; member
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h723 BroadcastableOpCategory broadcast_category; member
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1073 op_params.broadcast_category = in BroadcastAddFivefold()
Doptimized_ops.h2968 unswitched_params.broadcast_category == in BroadcastAddFivefold()
3362 unswitched_params.broadcast_category == in BroadcastMulFivefold()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h3081 unswitched_params.broadcast_category == in BroadcastAddFivefold()
3438 unswitched_params.broadcast_category == in BroadcastMulFivefold()