Searched refs:ElementwiseOptimizedPath (Results 1 – 5 of 5) sorted by relevance
39 enum class ElementwiseOptimizedPath { enum109 inline ElementwiseOptimizedPath select_broadcast_optimized_path( in select_broadcast_optimized_path()121 return ElementwiseOptimizedPath::kBroadcast2dBy1d; in select_broadcast_optimized_path()125 return ElementwiseOptimizedPath::kBroadcast2dBy1dReverseArguments; in select_broadcast_optimized_path()134 return ElementwiseOptimizedPath::kBroadcastNdByNd; in select_broadcast_optimized_path()136 return ElementwiseOptimizedPath::kBroadcastNdByNdReverseArguments; in select_broadcast_optimized_path()142 return ElementwiseOptimizedPath::kBroadcastLastDimReverseArguments; in select_broadcast_optimized_path()144 return ElementwiseOptimizedPath::kBroadcastLastDim; in select_broadcast_optimized_path()147 return ElementwiseOptimizedPath::kNone; in select_broadcast_optimized_path()151 ElementwiseOptimizedPath inline select_optimized_path( in select_optimized_path()[all …]
76 const ElementwiseOptimizedPath selected_optimized_path) { in handle_last_dim_broadcast()81 ElementwiseOptimizedPath::kBroadcastLastDimReverseArguments) { in handle_last_dim_broadcast()115 const ElementwiseOptimizedPath selected_optimized_path) { in handle_broadcast_mul()117 ElementwiseOptimizedPath::kBroadcastLastDim) || in handle_broadcast_mul()119 ElementwiseOptimizedPath::kBroadcastLastDimReverseArguments)) { in handle_broadcast_mul()127 ElementwiseOptimizedPath::kBroadcast2dBy1dReverseArguments) || in handle_broadcast_mul()129 ElementwiseOptimizedPath::kBroadcastNdByNdReverseArguments)) { in handle_broadcast_mul()136 ElementwiseOptimizedPath::kBroadcast2dBy1d) || in handle_broadcast_mul()138 ElementwiseOptimizedPath::kBroadcastNdByNd)); in handle_broadcast_mul()152 if ((selected_optimized_path == ElementwiseOptimizedPath::kBroadcastNdByNd) || in handle_broadcast_mul()[all …]
103 if (selected_optimized_path == ElementwiseOptimizedPath::kTreatAs1d) { in opt_div_out()122 } else if (selected_optimized_path != ElementwiseOptimizedPath::kNone) { in opt_div_out()126 ElementwiseOptimizedPath::kBroadcast2dBy1dReverseArguments) { in opt_div_out()133 ElementwiseOptimizedPath::kBroadcast2dBy1d); in opt_div_out()147 ElementwiseOptimizedPath::kBroadcast2dBy1dReverseArguments) { in opt_div_out()
142 if (selected_optimized_path == ElementwiseOptimizedPath::kTreatAs1d) { in opt_sub_out()165 } else if (selected_optimized_path != ElementwiseOptimizedPath::kNone) { in opt_sub_out()169 ElementwiseOptimizedPath::kBroadcast2dBy1dReverseArguments) { in opt_sub_out()176 ElementwiseOptimizedPath::kBroadcast2dBy1d); in opt_sub_out()194 ElementwiseOptimizedPath::kBroadcast2dBy1dReverseArguments) { in opt_sub_out()
119 if (selected_optimized_path == ElementwiseOptimizedPath::kTreatAs1d) { in opt_add_out()142 } else if (selected_optimized_path != ElementwiseOptimizedPath::kNone) { in opt_add_out()146 ElementwiseOptimizedPath::kBroadcast2dBy1dReverseArguments) { in opt_add_out()153 ElementwiseOptimizedPath::kBroadcast2dBy1d); in opt_add_out()