From bb27042bae9a01b4caee2f98a7ec421867690cb4 Mon Sep 17 00:00:00 2001 From: Gunhan Gulsoy Date: Fri, 7 Feb 2020 15:06:36 -0800 Subject: [PATCH] Automated rollback of commit 85bdc57c27ac894733dd161c2c809f15073d43be PiperOrigin-RevId: 293903584 Change-Id: Id110baa5604fc9def40b3a272abfa32daf8ae4eb --- tensorflow/python/BUILD | 7 ++++- .../tools/def_file_filter/symbols_pybind.txt | 31 +++++++++++++++++++ 2 files changed, 37 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 1aae82e89e2..23b2ff23daf 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -5759,6 +5759,7 @@ filegroup( "//tensorflow/core/grappler/clusters:single_machine", # tf_cluster "//tensorflow/core/grappler/clusters:utils", # tf_optimizer "//tensorflow/core/grappler/clusters:virtual_cluster", # tf_cluster + "//tensorflow/core/grappler/costs:analytical_cost_estimator", # cost analyzer "//tensorflow/core/grappler/costs:graph_memory", # tf_cluster "//tensorflow/core/grappler/costs:graph_properties", # tf_item "//tensorflow/core/grappler/costs:measuring_cost_estimator", # tf_cluster @@ -5778,7 +5779,11 @@ filegroup( "//tensorflow/python/eager:pywrap_tfe_lib", # pywrap_tfe_lib "//tensorflow/stream_executor:stream_executor_pimpl", # stat_summarizer "//tensorflow/tools/graph_transforms:transform_graph_lib", # transform_graph - ], + ] + if_xla_available([ + "//tensorflow/compiler/aot:tfcompile_lib", # tfcompile + "//tensorflow/compiler/xla:status_macros", # tfcompile + "//tensorflow/compiler/xla/service:hlo", # tfcompile + ]), visibility = ["//visibility:private"], ) diff --git a/tensorflow/tools/def_file_filter/symbols_pybind.txt b/tensorflow/tools/def_file_filter/symbols_pybind.txt index b202b3a81b2..da2e659c257 100644 --- a/tensorflow/tools/def_file_filter/symbols_pybind.txt +++ b/tensorflow/tools/def_file_filter/symbols_pybind.txt @@ -307,3 +307,34 @@ tensorflow::ProfilerSession::Create tensorflow::ProfilerSession::SerializeToString tensorflow::ProfilerSession::Status tensorflow::ProfilerSession::~ProfilerSession + +[status_macros] # tfcompile +xla::status_macros::MakeErrorStream::Impl::Impl +xla::status_macros::MakeErrorStream::Impl::~Impl +xla::status_macros::MakeErrorStream::Impl::GetStatus +xla::status_macros::MakeErrorStream::CheckNotDone + +[hlo] # tfcompile +xla::DfsHloVisitorBase::SetVisited +xla::DfsHloVisitorBase::SetVisited +xla::HloComputation::Accept +xla::HloComputation::AcceptOrdered +xla::HloComputation::CollectUnreachableRoots +xla::HloInstruction::ToString +xla::HloInstruction::Accept +xla::HloInstruction::Visit + +[tfcompile_lib] # tfcompile +tensorflow::tfcompile::Main + +[model_analyzer_lib] # model_analyzer +tensorflow::grappler::ModelAnalyzer::GenerateReport +tensorflow::grappler::ModelAnalyzer::ModelAnalyzer + +[analytical_cost_estimator] # cost_analyzer +tensorflow::grappler::AnalyticalCostEstimator::Initialize +tensorflow::grappler::AnalyticalCostEstimator::PredictCosts + +[cost_analyzer_lib] # cost_analyzer +tensorflow::grappler::CostAnalyzer::CostAnalyzer +tensorflow::grappler::CostAnalyzer::GenerateReport