Searched refs:expand_dims_if_unstacked (Results 1 – 1 of 1) sorted by relevance
72 auto expand_dims_if_unstacked = in ExpandDimsForBroadcast() local108 *inputs = VectorizerInput({expand_dims_if_unstacked(inputs->at(0), rank_0), in ExpandDimsForBroadcast()109 expand_dims_if_unstacked(inputs->at(1), rank_1)}); in ExpandDimsForBroadcast()