Searched refs:MapImpl (Results 1 – 2 of 2) sorted by relevance
1340 StatusOr<std::unique_ptr<Literal>> MapImpl(HloInstruction* map) { in MapImpl() function in xla::HloEvaluator::TypedVisitor1381 TF_ASSIGN_OR_RETURN(parent_->evaluated_[map], MapImpl<bool>(map)); in HandleMap()1385 TF_ASSIGN_OR_RETURN(parent_->evaluated_[map], MapImpl<uint8>(map)); in HandleMap()1389 TF_ASSIGN_OR_RETURN(parent_->evaluated_[map], MapImpl<uint32>(map)); in HandleMap()1393 TF_ASSIGN_OR_RETURN(parent_->evaluated_[map], MapImpl<uint64>(map)); in HandleMap()1397 TF_ASSIGN_OR_RETURN(parent_->evaluated_[map], MapImpl<int8>(map)); in HandleMap()1401 TF_ASSIGN_OR_RETURN(parent_->evaluated_[map], MapImpl<int32>(map)); in HandleMap()1405 TF_ASSIGN_OR_RETURN(parent_->evaluated_[map], MapImpl<int64>(map)); in HandleMap()1410 MapImpl<Eigen::half>(map)); in HandleMap()1414 TF_ASSIGN_OR_RETURN(parent_->evaluated_[map], MapImpl<float>(map)); in HandleMap()[all …]
138 public enum MapImpl { enum in BenchmarkHelpers