Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc7782 EXPECT_TRUE(iterations->literal().GetFirstInteger()); in TEST_F()
7783 EXPECT_EQ(*iterations->literal().GetFirstInteger(), 4); in TEST_F()
7817 EXPECT_TRUE(iterations->literal().GetFirstInteger()); in TEST_F()
7818 EXPECT_EQ(*iterations->literal().GetFirstInteger(), 2); in TEST_F()
7864 EXPECT_TRUE(iterations->literal().GetFirstInteger()); in TEST_F()
7865 EXPECT_EQ(*iterations->literal().GetFirstInteger(), 4); in TEST_F()
7911 EXPECT_TRUE(iterations->literal().GetFirstInteger()); in TEST_F()
7912 EXPECT_EQ(*iterations->literal().GetFirstInteger(), 8); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h137 absl::optional<int64> GetFirstInteger() const;
Dliteral.cc241 absl::optional<int64> LiteralBase::GetFirstInteger() const { in GetFirstInteger() function in xla::LiteralBase
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc1121 return {LessThanCompareInfo{lhs, *rhs->literal().GetFirstInteger()}}; in TrySimplifyTautologicalCompare()
1127 return {LessThanCompareInfo{rhs, *lhs->literal().GetFirstInteger()}}; in TrySimplifyTautologicalCompare()
4549 dynamic_slice->operand(i + 1)->literal().GetFirstInteger(); in HandleDynamicSlice()
4684 slice_dim_start->literal().GetFirstInteger(); in HandleDynamicUpdateSlice()