Searched refs:ExpectArray (Results 1 – 1 of 1) sorted by relevance
55 Status ExpectArray(const Shape& shape, absl::string_view op_type) { in ExpectArray() function242 TF_RETURN_IF_ERROR(ExpectArray(shape, "operand of unary operation")); in InferUnaryOpShape()375 TF_RETURN_IF_ERROR(ExpectArray(*shape, "operand of concatenation")); in InferConcatOpShape()649 TF_RETURN_IF_ERROR(ExpectArray(lhs, "lhs of dot")); in InferDotOpShape()650 TF_RETURN_IF_ERROR(ExpectArray(rhs, "rhs of dot")); in InferDotOpShape()902 TF_RETURN_IF_ERROR(ExpectArray(lhs, "lhs of elementwise binary operation")); in InferElementwiseBinaryOpShape()903 TF_RETURN_IF_ERROR(ExpectArray(rhs, "rhs of elementwise binary operation")); in InferElementwiseBinaryOpShape()973 TF_RETURN_IF_ERROR(ExpectArray( in InferBinaryOpShape()975 TF_RETURN_IF_ERROR(ExpectArray( in InferBinaryOpShape()1124 TF_RETURN_IF_ERROR(ExpectArray(*arg_shapes[i], "operand of map")); in InferMapShape()[all …]