Searched refs:broadcast_lhs (Results 1 – 2 of 2) sorted by relevance
41 const bool broadcast_lhs = lhs_shape.dims() < rhs_shape.dims(); in Compile() local42 const TensorShape* min_rank_shape = broadcast_lhs ? &lhs_shape : &rhs_shape; in Compile()43 const TensorShape* max_rank_shape = broadcast_lhs ? &rhs_shape : &lhs_shape; in Compile()92 if (broadcast_lhs) { in Compile()
3139 const bool broadcast_lhs = lhs_rank < rhs_rank; in inferReturnTypes() local3140 RankedTensorType min_rank_ty = broadcast_lhs ? lhs_ty : rhs_ty; in inferReturnTypes()3141 RankedTensorType max_rank_ty = broadcast_lhs ? rhs_ty : lhs_ty; in inferReturnTypes()3165 if (broadcast_lhs) { in inferReturnTypes()