Home
last modified time | relevance | path

Searched refs:arg_literals (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h59 absl::Span<const Literal* const> arg_literals) { in Evaluate() argument
60 return Evaluate(*module.entry_computation(), arg_literals); in Evaluate()
64 absl::Span<const Literal> arg_literals) { in Evaluate() argument
65 return Evaluate(*module.entry_computation(), arg_literals); in Evaluate()
88 absl::Span<const Literal* const> arg_literals);
91 absl::Span<const Literal> arg_literals) { in Evaluate() argument
93 for (const auto& l : arg_literals) { in Evaluate()
Dhlo_evaluator.cc232 absl::Span<const Literal* const> arg_literals) { in Evaluate() argument
237 if (arg_literals.size() != computation.num_parameters()) { in Evaluate()
240 computation.num_parameters() == 1 ? "" : "s", arg_literals.size()); in Evaluate()
242 for (int64 i = 0; i < arg_literals.size(); ++i) { in Evaluate()
245 const auto& arg_shape = arg_literals[i]->shape(); in Evaluate()
257 for (const auto& literal_ptr : arg_literals) { in Evaluate()
1209 std::vector<const Literal*> arg_literals; in HandleCall() local
1210 arg_literals.reserve(operands.size()); in HandleCall()
1213 arg_literals.push_back(&arg_literal); in HandleCall()
1220 embedded_evaluator.Evaluate(*computation, arg_literals)); in HandleCall()
[all …]
Dhlo_evaluator_typed_visitor.h1592 std::vector<Literal> arg_literals; in MapImpl()
1593 arg_literals.reserve(operands.size()); in MapImpl()
1604 arg_literals.push_back(std::move(curr_val_literal)); in MapImpl()
1608 embedded_evaluator.Evaluate(*computation, arg_literals) in MapImpl()
1705 absl::InlinedVector<const Literal*, 1> arg_literals(num_args); in HandleReduce()
1708 arg_literals[i] = &parent_->GetEvaluatedLiteralFor(reduce->inputs()[i]); in HandleReduce()
1709 VLOG(3) << "HandleReduce arg_literal: " << arg_literals[i]->ToString(); in HandleReduce()
1717 const Shape& arg_shape = arg_literals[0]->shape(); in HandleReduce()
1776 GetAsDouble<ReturnT>(*arg_literals[0], input_index); in HandleReduce()
1779 ShapeUtil::ForEachIndex(arg_literals[0]->shape(), base, in HandleReduce()
[all …]
Dhlo_evaluator_test.cc62 absl::Span<const Literal* const> arg_literals = {}) { in Evaluate() argument
66 return evaluator_.Evaluate(*m_->entry_computation(), arg_literals);
73 HloModule* module, absl::Span<const Literal* const> arg_literals = {}) { in EvaluateWithModule() argument
77 return evaluator_.Evaluate(*module->entry_computation(), arg_literals)
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.cc88 std::vector<Literal> arg_literals; in ExecuteOnStream() local
93 arg_literals.push_back(std::move(arg_literal)); in ExecuteOnStream()
102 evaluator_->Evaluate(*computation, arg_literals)); in ExecuteOnStream()