zzos/tensorflow
1
0
Fork 0
mirror of https://github.com/tensorflow/tensorflow.git synced 2026-09-28 05:13:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Automated Code Change

Browse Source 
PiperOrigin-RevId: 987276596
This commit is contained in:
A. Unique TensorFlower
2026-09-25 03:46:03 -07:00
committed by TensorFlower Gardener
parent 3059318f88
commit 019a0569e9
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tensorflow/compiler/mlir/quantization/tensorflow/debugging/dump_tensor_op.cc
+5 -5
View File
@@ -63,10 +63,10 @@ REGISTER_OP("DumpTensor")
class DumpTensorOp : public OpKernel {
public:
explicit DumpTensorOp(OpKernelConstruction* ctx) : OpKernel(ctx) {
string log_dir_path;
string file_name;
string func_name;
string node_name;
std::string log_dir_path;
std::string file_name;
std::string func_name;
std::string node_name;
OP_REQUIRES_OK(ctx, ctx->GetAttr("log_dir_path", &log_dir_path));
OP_REQUIRES_OK(ctx, ctx->GetAttr("enabled", &enabled_));
OP_REQUIRES_OK(ctx, ctx->GetAttr("file_name", &file_name));
@@ -92,7 +92,7 @@ class DumpTensorOp : public OpKernel {
quant_unit_proto.set_func_name(func_name);
quant_unit_proto.set_node_name(node_name);
string quant_unit_path = io::JoinPath(log_dir_path, "quant_unit.pb");
std::string quant_unit_path = io::JoinPath(log_dir_path, "quant_unit.pb");
OP_REQUIRES_OK(
ctx, SaveSerializedProtoToFile(quant_unit_proto.SerializeAsString(),
quant_unit_path, ctx->env()));