Remove superfluous tracemes in compression_utils.
Now that we execute compression as a tensorflow op, a traceme will automatically be generated for the op, so we don't need a second traceme inside the compression util. PiperOrigin-RevId: 312776419 Change-Id: I709c89b6a7fafaf41dbc8a64c8c28025cd7cd287
This commit is contained in:
parent
8fc976574e
commit
6b715d7238
@ -29,7 +29,6 @@ cc_library(
|
||||
"//tensorflow/core:lib",
|
||||
"//tensorflow/core:lib_internal",
|
||||
"//tensorflow/core:protos_all_cc",
|
||||
"//tensorflow/core/profiler/lib:traceme",
|
||||
"@com_google_absl//absl/memory",
|
||||
],
|
||||
)
|
||||
|
@ -17,16 +17,12 @@ limitations under the License.
|
||||
#include "tensorflow/core/common_runtime/dma_helper.h"
|
||||
#include "tensorflow/core/framework/tensor.pb.h"
|
||||
#include "tensorflow/core/platform/snappy.h"
|
||||
#include "tensorflow/core/profiler/lib/traceme.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace data {
|
||||
|
||||
Status CompressElement(const std::vector<Tensor>& element,
|
||||
CompressedElement* out) {
|
||||
tensorflow::profiler::TraceMe activity(
|
||||
"CompressElement", tensorflow::profiler::TraceMeLevel::kInfo);
|
||||
|
||||
// Step 1: Determine the total uncompressed size. This requires serializing
|
||||
// non-memcopyable tensors, which we save to use again later.
|
||||
std::vector<TensorProto> non_memcpy_components;
|
||||
@ -79,8 +75,6 @@ Status CompressElement(const std::vector<Tensor>& element,
|
||||
|
||||
Status UncompressElement(const CompressedElement& compressed,
|
||||
std::vector<Tensor>* out) {
|
||||
tensorflow::profiler::TraceMe activity(
|
||||
"UncompressElement", tensorflow::profiler::TraceMeLevel::kInfo);
|
||||
int num_components = compressed.component_metadata_size();
|
||||
out->clear();
|
||||
out->reserve(num_components);
|
||||
|
Loading…
x
Reference in New Issue
Block a user