Home
last modified time | relevance | path

Searched refs:arg_literals_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h304 return *arg_literals_.at(hlo->parameter_number()); in GetEvaluatedLiteralFor()
355 std::vector<const Literal*> arg_literals_; variable
Dhlo_evaluator.cc259 arg_literals_.clear(); in Evaluate()
261 arg_literals_.push_back(&*literal_ptr); in Evaluate()
306 arg_literals_.clear(); in Evaluate()
455 CHECK_LT(parameter->parameter_number(), arg_literals_.size()); in HandleParameter()
458 const Literal* input_literal = arg_literals_[parameter->parameter_number()]; in HandleParameter()