Searched refs:right_leaf_rank (Results 1 – 1 of 1) sorted by relevance
3234 const int right_leaf_rank = (*ctx.op_child_input_props)[1].shape().dim_size(); in ConstantPushDownBiasAdd() local3236 if (left_leaf_rank != 1 && right_leaf_rank != 1) return false; in ConstantPushDownBiasAdd()3241 const int vector_rank = vector_idx == 0 ? left_leaf_rank : right_leaf_rank; in ConstantPushDownBiasAdd()