Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_typed_visitor.h178 const Literal& operand_literal = in HandleAbs() local
184 operand_literal))); in HandleAbs()
993 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand); in HandleReverse() local
1002 return operand_literal.Get<ReturnT>(from_index); in HandleReverse()
1473 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand); in HandleDynamicSlice() local
1480 operand_literal, in HandleDynamicSlice()
1488 operand_literal, in HandleDynamicSlice()
1496 operand_literal, in HandleDynamicSlice()
1504 operand_literal, in HandleDynamicSlice()
1537 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand); in HandleDynamicUpdateSlice() local
[all …]
Dhlo_evaluator.h311 const Literal& operand_literal) { in ElementWiseUnaryOpImpl() argument
319 return unary_op(operand_literal.Get<NativeT>(multi_index)); in ElementWiseUnaryOpImpl()
Dhlo_evaluator_test.cc1776 auto operand_literal = in TEST_P() local
1780 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
1809 auto operand_literal = in TEST_P() local
1813 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
1847 auto operand_literal = in TEST_P() local
1851 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
1883 auto operand_literal = in TEST_P() local
1887 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
2007 auto operand_literal = LiteralUtil::CreateR4FromArray4D<float>(input); in TEST_P() local
2009 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
Dhlo_evaluator.cc383 const Literal& operand_literal = GetEvaluatedLiteralFor(bitcast->operand(0)); in HandleBitcast() local
385 TF_RET_CHECK(operand_literal.size_bytes() == result.size_bytes()); in HandleBitcast()
386 memcpy(result.untyped_data(), operand_literal.untyped_data(), in HandleBitcast()
387 operand_literal.size_bytes()); in HandleBitcast()