HomeSort by relevance Sort by last modified time
    Searched refs:arg_literals (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator.h 52 // Precondition: The indices of arg_literals correspond to the parameter
59 absl::Span<const Literal* const> arg_literals) {
60 return Evaluate(*module.entry_computation(), arg_literals);
64 absl::Span<const Literal> arg_literals) {
65 return Evaluate(*module.entry_computation(), arg_literals);
70 // Precondition: The indices of arg_literals correspond to the parameter
88 absl::Span<const Literal* const> arg_literals);
91 absl::Span<const Literal> arg_literals) {
93 for (const auto& l : arg_literals) {
272 // is looked up in arg_literals
    [all...]
hlo_evaluator.cc 232 absl::Span<const Literal* const> arg_literals) {
237 if (arg_literals.size() != computation.num_parameters()) {
240 computation.num_parameters() == 1 ? "" : "s", arg_literals.size());
242 for (int64 i = 0; i < arg_literals.size(); ++i) {
245 const auto& arg_shape = arg_literals[i]->shape();
257 for (const auto& literal_ptr : arg_literals) {
1209 std::vector<const Literal*> arg_literals; local
1244 std::vector<const Literal*> arg_literals; local
    [all...]
hlo_evaluator_typed_visitor.h     [all...]
hlo_evaluator_test.cc 62 absl::Span<const Literal* const> arg_literals = {}) {
66 return evaluator_.Evaluate(*m_->entry_computation(), arg_literals);
73 HloModule* module, absl::Span<const Literal* const> arg_literals = {}) {
77 return evaluator_.Evaluate(*module->entry_computation(), arg_literals)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executable.cc 88 std::vector<Literal> arg_literals; local
93 arg_literals.push_back(std::move(arg_literal));
102 evaluator_->Evaluate(*computation, arg_literals));

Completed in 285 milliseconds