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