Searched refs:ExpectArray (Results 1 – 1 of 1) sorted by relevance
54 Status ExpectArray(const Shape& shape, absl::string_view op_type) { in ExpectArray() function262 TF_RETURN_IF_ERROR(ExpectArray(shape, "operand of unary operation")); in InferUnaryOpShape()394 TF_RETURN_IF_ERROR(ExpectArray(*shape, "operand of concatenation")); in InferConcatOpShape()652 TF_RETURN_IF_ERROR(ExpectArray(lhs, "lhs of dot")); in InferDotOpShape()653 TF_RETURN_IF_ERROR(ExpectArray(rhs, "rhs of dot")); in InferDotOpShape()898 TF_RETURN_IF_ERROR(ExpectArray(lhs, "lhs of elementwise binary operation")); in InferElementwiseBinaryOpShape()899 TF_RETURN_IF_ERROR(ExpectArray(rhs, "rhs of elementwise binary operation")); in InferElementwiseBinaryOpShape()969 TF_RETURN_IF_ERROR(ExpectArray( in InferBinaryOpShape()971 TF_RETURN_IF_ERROR(ExpectArray( in InferBinaryOpShape()1126 TF_RETURN_IF_ERROR(ExpectArray(*arg_shapes[i], "operand of map")); in InferMapShape()[all …]