Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_control_flow_flattening.cc101 value = *condition->operand(1)->literal().GetFirstInteger(); in GetLoopBound()
106 value = *condition->operand(0)->literal().GetFirstInteger(); in GetLoopBound()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc234 loop_cond_root->operand(0)->literal().GetFirstInteger(); in PatternMatchLoopCondComparison()
250 loop_cond_root->operand(1)->literal().GetFirstInteger(); in PatternMatchLoopCondComparison()
317 .GetFirstInteger(); in PatternMatchInductionVarUpdate()
336 .GetFirstInteger(); in PatternMatchInductionVarUpdate()
355 .GetFirstInteger(); in PatternMatchInductionVarUpdate()
449 return DynamicOrStaticValue{value->GetFirstInteger()}; in EvaluateWhileLoopParamInitValue()
Dinstruction_fusion.cc939 return operand->literal().GetFirstInteger(); in ShouldFuseInPlaceOp()
Dalgebraic_simplifier.cc976 return {LessThanCompareInfo{lhs, *rhs->literal().GetFirstInteger()}}; in TrySimplifyTautologicalCompare()
982 return {LessThanCompareInfo{rhs, *lhs->literal().GetFirstInteger()}}; in TrySimplifyTautologicalCompare()
4919 dynamic_slice->operand(i + 1)->literal().GetFirstInteger(); in HandleDynamicSlice()
5080 slice_dim_start->literal().GetFirstInteger(); in HandleDynamicUpdateSlice()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h138 std::optional<int64_t> GetFirstInteger() const;
Dliteral.cc343 std::optional<int64_t> LiteralBase::GetFirstInteger() const { in GetFirstInteger() function in xla::LiteralBase
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc9831 EXPECT_TRUE(iterations->literal().GetFirstInteger()); in TEST_F()
9832 EXPECT_EQ(*iterations->literal().GetFirstInteger(), 4); in TEST_F()
9866 EXPECT_TRUE(iterations->literal().GetFirstInteger()); in TEST_F()
9867 EXPECT_EQ(*iterations->literal().GetFirstInteger(), 2); in TEST_F()
9913 EXPECT_TRUE(iterations->literal().GetFirstInteger()); in TEST_F()
9914 EXPECT_EQ(*iterations->literal().GetFirstInteger(), 4); in TEST_F()
9960 EXPECT_TRUE(iterations->literal().GetFirstInteger()); in TEST_F()
9961 EXPECT_EQ(*iterations->literal().GetFirstInteger(), 8); in TEST_F()