Searched refs:ExpectArray (Results 1 – 1 of 1) sorted by relevance
52 Status ExpectArray(const Shape& shape, absl::string_view op_type) { in ExpectArray() function238 TF_RETURN_IF_ERROR(ExpectArray(shape, "operand of unary operation")); in InferUnaryOpShape()360 TF_RETURN_IF_ERROR(ExpectArray(*shape, "operand of concatenation")); in InferConcatOpShape()611 TF_RETURN_IF_ERROR(ExpectArray(lhs, "lhs of dot")); in InferDotOpShape()612 TF_RETURN_IF_ERROR(ExpectArray(rhs, "rhs of dot")); in InferDotOpShape()863 TF_RETURN_IF_ERROR(ExpectArray(lhs, "lhs of elementwise binary operation")); in InferElementwiseBinaryOpShape()864 TF_RETURN_IF_ERROR(ExpectArray(rhs, "rhs of elementwise binary operation")); in InferElementwiseBinaryOpShape()925 TF_RETURN_IF_ERROR(ExpectArray( in InferBinaryOpShape()927 TF_RETURN_IF_ERROR(ExpectArray( in InferBinaryOpShape()1082 TF_RETURN_IF_ERROR(ExpectArray(*arg_shapes[i], "operand of map")); in InferMapShape()[all …]