Home
last modified time | relevance | path

Searched refs:ExpectArray (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc54 Status ExpectArray(const Shape& shape, absl::string_view op_type) { in ExpectArray() function
262 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 …]