Home
last modified time | relevance | path

Searched refs:evaluated_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc255 evaluated_.clear(); in Evaluate()
288 evaluated_.clear(); in Evaluate()
388 evaluated_[bitcast] = std::move(result); in HandleBitcast()
404 evaluated_[get_dimension_size] = in HandleGetDimensionSize()
413 evaluated_[get_dimension_size] = std::move(output); in HandleGetDimensionSize()
438 evaluated_[reshape], in HandleReshape()
445 evaluated_[transpose] = GetEvaluatedLiteralFor(transpose->operand(0)) in HandleTranspose()
487 evaluated_[concatenate] = std::move(result_literal); in HandleConcatenate()
526 TF_ASSIGN_OR_RETURN(evaluated_[is_finite], std::move(result_or)); in HandleIsFinite()
536 TF_ASSIGN_OR_RETURN(evaluated_[is_finite], std::move(result_or)); in HandleIsFinite()
[all …]
Dhlo_evaluator_typed_visitor.h156 TF_ASSIGN_OR_RETURN(parent_->evaluated_[abs], in HandleAbs()
167 TF_ASSIGN_OR_RETURN(parent_->evaluated_[abs], in HandleAbs()
181 parent_->evaluated_[abs], in HandleAbs()
206 parent_->evaluated_[round], in HandleRound()
228 TF_ASSIGN_OR_RETURN(parent_->evaluated_[ceil], in HandleCeil()
252 parent_->evaluated_[convert] = std::move(result); in HandleConvert()
263 parent_->evaluated_[convert] = std::move(result); in HandleBitcastConvert()
268 TF_ASSIGN_OR_RETURN(parent_->evaluated_[exp], in HandleExp()
280 parent_->evaluated_[expm1], in HandleExpm1()
303 parent_->evaluated_[floor], in HandleFloor()
[all …]
Dhlo_evaluator.h282 auto it = evaluated_.find(hlo); in GetEvaluatedLiteralFor()
283 CHECK(it != evaluated_.end()) in GetEvaluatedLiteralFor()
301 absl::node_hash_map<const HloInstruction*, Literal> evaluated_; variable