Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h61 absl::Span<const Literal* const> arg_literals) { in Evaluate() argument
62 return Evaluate(*module.entry_computation(), arg_literals); in Evaluate()
66 absl::Span<const Literal> arg_literals) { in Evaluate() argument
67 return Evaluate(*module.entry_computation(), arg_literals); in Evaluate()
90 absl::Span<const Literal* const> arg_literals);
93 absl::Span<const Literal> arg_literals) { in Evaluate() argument
95 for (const auto& l : arg_literals) { in Evaluate()
Dhlo_evaluator.cc234 absl::Span<const Literal* const> arg_literals) { in Evaluate() argument
239 if (arg_literals.size() != computation.num_parameters()) { in Evaluate()
242 computation.num_parameters() == 1 ? "" : "s", arg_literals.size()); in Evaluate()
244 for (int64 i = 0; i < arg_literals.size(); ++i) { in Evaluate()
247 const auto& arg_shape = arg_literals[i]->shape(); in Evaluate()
260 for (const auto& literal_ptr : arg_literals) { in Evaluate()
1921 std::vector<const Literal*> arg_literals; in HandleCall() local
1922 arg_literals.reserve(operands.size()); in HandleCall()
1925 arg_literals.push_back(&arg_literal); in HandleCall()
1932 embedded_evaluator.Evaluate(*computation, arg_literals)); in HandleCall()
[all …]
Dhlo_evaluator_typed_visitor.h1736 std::vector<Literal> arg_literals;
1737 arg_literals.reserve(operands.size());
1748 arg_literals.push_back(std::move(curr_val_literal));
1752 embedded_evaluator.Evaluate(*computation, arg_literals)
Dhlo_evaluator_test.cc63 absl::Span<const Literal* const> arg_literals = {}) { in Evaluate() argument
67 return evaluator_.Evaluate(*m_->entry_computation(), arg_literals);
74 HloModule* module, absl::Span<const Literal* const> arg_literals = {}) { in EvaluateWithModule() argument
78 return evaluator_.Evaluate(*module->entry_computation(), arg_literals)
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.cc56 const HloComputation& computation, absl::Span<const Literal> arg_literals) { in Evaluate() argument
60 return evaluator_->Evaluate(computation, arg_literals); in Evaluate()
Dexecutable_base.cc110 std::vector<Literal> arg_literals; in ExecuteAsyncOnStream() local
120 arg_literals.push_back(std::move(arg_literal)); in ExecuteAsyncOnStream()
124 Evaluate(run_options, *computation, arg_literals)); in ExecuteAsyncOnStream()
Dexecutable_base.h49 absl::Span<const Literal> arg_literals) = 0;
Dexecutable.h56 absl::Span<const Literal> arg_literals) override