Searched refs:operand_subshape (Results 1 – 2 of 2) sorted by relevance
823 const Shape& operand_subshape = ShapeUtil::GetSubshape( in HandleCustomCall() local826 TF_RET_CHECK(operand_subshape == output_subshape) in HandleCustomCall()827 << "Different aliasing shapes: " << operand_subshape.ToString() in HandleCustomCall()830 TF_RET_CHECK(ShapeUtil::Compatible(output_subshape, operand_subshape)) in HandleCustomCall()831 << "Different aliasing shapes: " << operand_subshape.ToString() in HandleCustomCall()
1269 const Shape& operand_subshape = in CanShareOperandBufferWithUser() local1275 if (!ShapeUtil::Equal(operand_subshape, user_subshape)) { in CanShareOperandBufferWithUser()