Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc373 return LiteralUtil::CreateR0<bool>(literal.GetFirstElement<bool>()); in GetFirstScalarLiteral()
376 return LiteralUtil::CreateR0<int8>(literal.GetFirstElement<int8>()); in GetFirstScalarLiteral()
378 return LiteralUtil::CreateR0<uint8>(literal.GetFirstElement<uint8>()); in GetFirstScalarLiteral()
382 literal.GetFirstElement<bfloat16>()); in GetFirstScalarLiteral()
384 return LiteralUtil::CreateR0<half>(literal.GetFirstElement<half>()); in GetFirstScalarLiteral()
386 return LiteralUtil::CreateR0<int16>(literal.GetFirstElement<int16>()); in GetFirstScalarLiteral()
388 return LiteralUtil::CreateR0<uint16>(literal.GetFirstElement<uint16>()); in GetFirstScalarLiteral()
391 return LiteralUtil::CreateR0<float>(literal.GetFirstElement<float>()); in GetFirstScalarLiteral()
393 return LiteralUtil::CreateR0<int32>(literal.GetFirstElement<int32>()); in GetFirstScalarLiteral()
395 return LiteralUtil::CreateR0<uint32>(literal.GetFirstElement<uint32>()); in GetFirstScalarLiteral()
[all …]
Dliteral.h123 NativeT GetFirstElement() const;
945 NativeT LiteralBase::GetFirstElement() const { in GetFirstElement() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_analysis.cc177 return l.GetFirstElement<int8>(); in LiteralAsScalarInt64()
179 return l.GetFirstElement<int16>(); in LiteralAsScalarInt64()
181 return l.GetFirstElement<int32>(); in LiteralAsScalarInt64()
183 return l.GetFirstElement<int64>(); in LiteralAsScalarInt64()
185 return l.GetFirstElement<uint8>(); in LiteralAsScalarInt64()
187 return l.GetFirstElement<uint16>(); in LiteralAsScalarInt64()
189 return l.GetFirstElement<uint32>(); in LiteralAsScalarInt64()
191 uint64 v = l.GetFirstElement<uint64>(); in LiteralAsScalarInt64()
484 bool cond_returns_true = cond_result_pred.GetFirstElement<bool>(); in ComputeWhileLoopTripCountUpperBound()
Dhlo_constant_folding_test.cc61 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<int64>(), in TEST_F()
83 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F()
235 .GetFirstElement<int32>()); in TEST_F()
Dcall_inliner_test.cc70 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F()
75 EXPECT_EQ(prior->literal().GetFirstElement<float>(), 24); in TEST_F()
Dalgebraic_simplifier.cc94 return static_cast<double>(c->literal().GetFirstElement<bfloat16>()); in IsAllFpConstantPowerOf2()
96 return static_cast<double>(c->literal().GetFirstElement<Eigen::half>()); in IsAllFpConstantPowerOf2()
98 return c->literal().GetFirstElement<float>(); in IsAllFpConstantPowerOf2()
100 return c->literal().GetFirstElement<double>(); in IsAllFpConstantPowerOf2()
869 int64 b_value = c->literal().GetFirstElement<T>(); in TryDivideToShift()
911 uint64 b_value = c->literal().GetFirstElement<T>(); in TryDivideToShift()
2471 int64 b_value = c->literal().GetFirstElement<T>(); in TryRemainderToAnd()
2510 uint64 b_value = c->literal().GetFirstElement<T>(); in TryRemainderToAnd()
Dhlo_evaluator_test.cc3022 EXPECT_EQ(actual[0].GetFirstElement<int32>(), static_cast<int32>(pow31)); in TEST_F()
3023 EXPECT_EQ(actual[1].GetFirstElement<int32>(), in TEST_F()
3025 EXPECT_EQ(actual[2].GetFirstElement<int32>(), in TEST_F()
3059 EXPECT_EQ(actual.GetFirstElement<uint32>(), static_cast<uint32>(3)); in TEST_F()
Dalgebraic_simplifier_test.cc567 EXPECT_EQ(3.14f, root->operand(0)->literal().GetFirstElement<float>()); in TEST_F()
1234 EXPECT_EQ(root->literal().GetFirstElement<float>(), 1); in TEST_F()
1263 EXPECT_EQ(root->operand(0)->literal().GetFirstElement<float>(), 1); in TEST_F()
1336 EXPECT_EQ(root->operand(0)->operand(0)->literal().GetFirstElement<float>(), in TEST_F()
2441 EXPECT_EQ(0.0f, root->operand(0)->literal().GetFirstElement<float>()); in TEST_F()
Dhlo_evaluator_typed_visitor.h2866 parent_->GetEvaluatedLiteralFor(index).GetFirstElement<IndexT>()); in DynamicSlice()
2901 parent_->GetEvaluatedLiteralFor(index).GetFirstElement<IndexT>()); in DynamicUpdateSlice()
Dhlo_evaluator.cc1338 keep_going = cond_val.GetFirstElement<bool>(); in HandleWhile()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsequence_ops.cc153 float start = start_literal.GetFirstElement<float>(); in Compile()
154 float stop = stop_literal.GetFirstElement<float>(); in Compile()
169 double start = start_literal.GetFirstElement<double>(); in Compile()
170 double stop = stop_literal.GetFirstElement<double>(); in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc184 return alpha.GetFirstElement<double>(); in TryRewriteToCudnnForwardRelu()