File tree Expand file tree Collapse file tree 3 files changed +18
-4
lines changed Expand file tree Collapse file tree 3 files changed +18
-4
lines changed Original file line number Diff line number Diff line change @@ -2733,7 +2733,6 @@ tf_cc_test_mkl(
2733
2733
"//tensorflow/core:test_main" ,
2734
2734
"//tensorflow/core:testlib" ,
2735
2735
"//tensorflow/core/util:onednn_env_vars" ,
2736
- "@com_google_absl//absl/status" ,
2737
2736
],
2738
2737
)
2739
2738
Original file line number Diff line number Diff line change @@ -17,7 +17,6 @@ limitations under the License.
17
17
18
18
#include " tensorflow/core/common_runtime/mkl_cpu_allocator.h"
19
19
20
- #include " absl/status/status.h"
21
20
#include " tensorflow/core/lib/core/status_test_util.h"
22
21
#include " tensorflow/core/platform/logging.h"
23
22
#include " tensorflow/core/platform/test.h"
@@ -42,10 +41,10 @@ TEST(MKLBFCAllocatorTest, TestMaxLimit) {
42
41
EXPECT_EQ (stats->bytes_limit , max_mem_bytes);
43
42
44
43
setenv (MklCPUAllocator::kMaxLimitStr , " wrong-input" , 1 );
45
- EXPECT_TRUE (absl ::IsInvalidArgument (a.Initialize ()));
44
+ EXPECT_TRUE (errors ::IsInvalidArgument (a.Initialize ()));
46
45
47
46
setenv (MklCPUAllocator::kMaxLimitStr , " -20" , 1 );
48
- EXPECT_TRUE (absl ::IsInvalidArgument (a.Initialize ()));
47
+ EXPECT_TRUE (errors ::IsInvalidArgument (a.Initialize ()));
49
48
}
50
49
51
50
} // namespace tensorflow
Original file line number Diff line number Diff line change @@ -63,6 +63,22 @@ using tsl::errors::InternalWithPayloads;
63
63
using tsl::errors::InvalidArgument;
64
64
using tsl::errors::InvalidArgumentWithPayloads;
65
65
using tsl::errors::IOError;
66
+ using tsl::errors::IsAborted;
67
+ using tsl::errors::IsAlreadyExists;
68
+ using tsl::errors::IsCancelled;
69
+ using tsl::errors::IsDataLoss;
70
+ using tsl::errors::IsDeadlineExceeded;
71
+ using tsl::errors::IsFailedPrecondition;
72
+ using tsl::errors::IsInternal;
73
+ using tsl::errors::IsInvalidArgument;
74
+ using tsl::errors::IsNotFound;
75
+ using tsl::errors::IsOutOfRange;
76
+ using tsl::errors::IsPermissionDenied;
77
+ using tsl::errors::IsResourceExhausted;
78
+ using tsl::errors::IsUnauthenticated;
79
+ using tsl::errors::IsUnavailable;
80
+ using tsl::errors::IsUnimplemented;
81
+ using tsl::errors::IsUnknown;
66
82
using tsl::errors::NotFound;
67
83
using tsl::errors::NotFoundWithPayloads;
68
84
using tsl::errors::OutOfRange;
You can’t perform that action at this time.
0 commit comments