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.h211 const Literal& operand_literal =
218 operand_literal)));
1070 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand);
1079 return operand_literal.Get<ReturnT>(from_index);
1617 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand);
1624 operand_literal,
1632 operand_literal,
1640 operand_literal,
1648 operand_literal,
1681 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand);
[all …]
Dhlo_evaluator.h335 const Literal& operand_literal) { in ElementWiseUnaryOpImpl() argument
343 return unary_op(operand_literal.Get<NativeT>(multi_index)); in ElementWiseUnaryOpImpl()
Dhlo_evaluator.cc403 const Literal& operand_literal = GetEvaluatedLiteralFor(bitcast->operand(0)); in HandleBitcast() local
405 TF_RET_CHECK(operand_literal.size_bytes() == result.size_bytes()); in HandleBitcast()
406 memcpy(result.untyped_data(), operand_literal.untyped_data(), in HandleBitcast()
407 operand_literal.size_bytes()); in HandleBitcast()
439 const Literal& operand_literal = in HandleSetDimensionSize() local
442 memcpy(result.untyped_data(), operand_literal.untyped_data(), in HandleSetDimensionSize()
443 operand_literal.size_bytes()); in HandleSetDimensionSize()
Dhlo_evaluator_test.cc2979 auto operand_literal = in TEST_P() local
2983 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
3012 auto operand_literal = in TEST_P() local
3016 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
3050 auto operand_literal = in TEST_P() local
3054 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
3086 auto operand_literal = in TEST_P() local
3090 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()
3210 auto operand_literal = LiteralUtil::CreateR4FromArray4D<float>(input); in TEST_P() local
3212 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()