Searched refs:status_or_shape (Results 1 – 3 of 3) sorted by relevance
157 StatusOr<Shape> status_or_shape; in TEST() local160 status_or_shape = in TEST()163 EXPECT_EQ(status_or_shape.status().code(), in TEST()167 status_or_shape = in TEST()169 EXPECT_TRUE(status_or_shape.ok()); in TEST()174 status_or_shape = in TEST()177 EXPECT_TRUE(status_or_shape.ok()); in TEST()
345 auto status_or_shape = pooling.builder()->GetShape(pooling); in Compile() local346 OP_REQUIRES_OK(ctx, status_or_shape.status()); in Compile()347 OP_REQUIRES_OK(ctx, XLAShapeToTensorShape(status_or_shape.ValueOrDie(), in Compile()
776 StatusOr<const Shape> status_or_shape = ShapeInference::InferTernaryOpShape( in TernaryOp() local778 if (!status_or_shape.status().ok()) { in TernaryOp()781 status_or_shape.status().error_message()); in TernaryOp()784 return AddOpWithShape(triop, status_or_shape.ValueOrDie(), in TernaryOp()