experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

added includes for each symbol used in summary_op

Browse Source
This commit is contained in:
Daniel Nguyen 2020-07-27 22:12:15 +00:00
parent ba6d661ba8
commit 64d3dbdba3
4 changed files with 35 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/c/kernels/BUILD
View File

@ -31,7 +31,9 @@ tf_kernel_library(
"//tensorflow/c/kernels:tensor_shape_utils",
"//tensorflow/c:kernels",
"//tensorflow/c:tf_tensor",
"//tensorflow/core:framework"
"//tensorflow/c:tf_status",
"//tensorflow/core:framework",
"//tensorflow/core:lib",
],
)
@ -51,6 +53,7 @@ tf_gen_op_libs(
op_lib_names = ["summary"],
deps = [
"//tensorflow/c:ops",
"//tensorflow/c:tf_status",
"//tensorflow/core:lib",
],
)
@ -73,7 +76,10 @@ tf_cc_test(
srcs = ["summary_op_test.cc"],
deps = [
":summary_op",
":summary_op_lib",
"//tensorflow/c:kernels",
"//tensorflow/core:framework",
"//tensorflow/core:lib",
"//tensorflow/core:test",
"//tensorflow/core:test_main",
"//tensorflow/core:testlib"
],
@ -94,6 +100,7 @@ tf_cc_test(
":tensor_shape_utils",
"//tensorflow/core:lib",
"//tensorflow/core:test",
"//tensorflow/core:framework",
"//tensorflow/core:test_main",
],
)

1
tensorflow/c/kernels/ops/summary.cc
View File

@ -14,6 +14,7 @@ limitations under the License.
==============================================================================*/
#include "tensorflow/c/ops.h"
#include "tensorflow/c/tf_status.h"
#include "tensorflow/core/framework/selective_registration.h"
#include "tensorflow/core/platform/logging.h"

13
tensorflow/c/kernels/summary_op.cc
View File

@ -15,10 +15,17 @@ limitations under the License.
==============================================================================*/
#include <sstream>
#include <string>
#include "tensorflow/c/kernels/tensor_shape_utils.h"
#include "tensorflow/c/kernels.h"
#include "tensorflow/c/tf_tensor.h"
#include "tensorflow/c/kernels/tensor_shape_utils.h"
#include "tensorflow/c/tf_status.h"
#include "tensorflow/core/platform/protobuf.h"
#include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/platform/macros.h"
#include "tensorflow/core/platform/tstring.h"
#include "tensorflow/core/platform/strcat.h"
#include "tensorflow/core/framework/selective_registration.h"
#include "tensorflow/core/framework/summary.pb.h"
#include "tensorflow/core/framework/types.h"
@ -60,7 +67,7 @@ void ScalarSummaryOp_Delete(void* kernel) {
bool IsSameSize(TF_Tensor* tensor1, TF_Tensor* tensor2);
// Returns a string representation of a single tag or empty string if there
// are multiple tags
tensorflow::string SingleTag(TF_Tensor* tags);
std::string SingleTag(TF_Tensor* tags);
template<typename T>
void ScalarSummaryOp_Compute(void* kernel, TF_OpKernelContext* ctx) {
@ -117,7 +124,7 @@ bool IsSameSize(TF_Tensor* tensor1, TF_Tensor* tensor2) {
return true;
}
tensorflow::string SingleTag(TF_Tensor* tags) {
std::string SingleTag(TF_Tensor* tags) {
if (TF_TensorElementCount(tags) == 1) {
const char* single_tag = static_cast<tensorflow::tstring*>(
TF_TensorData(tags))->c_str();

16
tensorflow/c/kernels/summary_op_test.cc
View File

@ -13,12 +13,26 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include "tensorflow/c/kernels.h"
#include "tensorflow/core/framework/allocator.h"
#include "tensorflow/core/framework/device_base.h"
#include "tensorflow/core/framework/attr_value.pb.h"
#include "tensorflow/core/framework/attr_value_util.h"
#include "tensorflow/core/framework/node_def.pb.h"
#include "tensorflow/core/framework/op_kernel.h"
#include "tensorflow/core/platform/test.h"
#include "tensorflow/core/framework/types.h"
#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/op_def_builder.h"
#include "tensorflow/core/framework/summary.pb.h"
#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/status.h"
#include "tensorflow/core/platform/strcat.h"
#include "tensorflow/core/platform/test.h"
#include "tensorflow/core/platform/protobuf.h"
#include "tensorflow/core/protobuf/error_codes.pb.h"
#include "tensorflow/core/lib/gtl/inlined_vector.h"
namespace tensorflow {
namespace {