Searched refs:arg_literals (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.h | 103 absl::Span<const Literal* const> arg_literals) { in Evaluate() argument
104 return Evaluate(*module.entry_computation(), arg_literals); in Evaluate()
108 absl::Span<const Literal> arg_literals) { in Evaluate() argument
109 return Evaluate(*module.entry_computation(), arg_literals); in Evaluate()
132 absl::Span<const Literal* const> arg_literals);
135 absl::Span<const Literal> arg_literals) { in Evaluate() argument
137 for (const auto& l : arg_literals) { in Evaluate()
|
| D | hlo_evaluator.cc | 773 absl::Span<const Literal* const> arg_literals) { in Evaluate() argument
778 if (arg_literals.size() != computation.num_parameters()) { in Evaluate()
781 computation.num_parameters() == 1 ? "" : "s", arg_literals.size()); in Evaluate()
783 for (int64_t i = 0; i < arg_literals.size(); ++i) { in Evaluate()
786 const auto& arg_shape = arg_literals[i]->shape(); in Evaluate()
799 for (const auto& literal_ptr : arg_literals) { in Evaluate()
3035 std::vector<const Literal*> arg_literals; in HandleAsyncStart() local
3036 arg_literals.reserve(async_start->operands().size()); in HandleAsyncStart()
3039 arg_literals.push_back(&arg_literal); in HandleAsyncStart()
3048 arg_literals)); in HandleAsyncStart()
[all …]
|
| D | hlo_evaluator_typed_visitor.h | 1551 std::vector<Literal> arg_literals;
1552 arg_literals.reserve(operands.size());
1559 arg_literals.push_back(
1564 embedded_evaluator.Evaluate(*computation, arg_literals).value();
|
| D | hlo_evaluator_test.cc | 61 absl::Span<const Literal* const> arg_literals = {}) { in Evaluate() argument
65 return evaluator_.Evaluate(*m_->entry_computation(), arg_literals);
72 HloModule* module, absl::Span<const Literal* const> arg_literals = {}) { in EvaluateWithModule() argument
76 return evaluator_.Evaluate(*module->entry_computation(), arg_literals)
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 56 const HloComputation& computation, absl::Span<const Literal> arg_literals) { in Evaluate() argument
60 return evaluator_->Evaluate(computation, arg_literals); in Evaluate()
|
| D | executable_base.cc | 110 std::vector<Literal> arg_literals; in ExecuteAsyncOnStream() local
112 arg_literals.reserve(num_parameters); in ExecuteAsyncOnStream()
122 arg_literals.push_back(std::move(arg_literal)); in ExecuteAsyncOnStream()
126 Evaluate(run_options, *computation, arg_literals)); in ExecuteAsyncOnStream()
|
| D | executable.h | 53 absl::Span<const Literal> arg_literals) override
|
| D | executable_base.h | 49 absl::Span<const Literal> arg_literals) = 0;
|