TensorFlow: Doc and linter fixes, some additional tests and
error handling, updates to website. Changes: - Removes redundant reshape from image models by @mrry - Default TensorBoard to localhost by @danmane - Reformatting of tensorflow/core by @josh11b - Make tutorials backwards compatible to 0.5.0 by @girving - Improve print documentation (md files not updated). - Add proper scrolling to sitemap by @martinwicke Base CL: 107956254
This commit is contained in:
Vijay Vasudevan
parent
cb9fa5fc9d
commit
56313def00
@ -1,7 +1,7 @@
|
||||
#include "tensorflow/core/common_runtime/device_set.h"
|
||||
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/public/status.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -379,8 +379,8 @@ void DirectSession::RestoreStatefulNodes(Graph* graph) {
|
||||
}
|
||||
}
|
||||
|
||||
Status DirectSession::CreateGraphs(gtl::ArraySlice<string> feeds,
|
||||
gtl::ArraySlice<string> fetches,
|
||||
Status DirectSession::CreateGraphs(
|
||||
gtl::ArraySlice<string> feeds, gtl::ArraySlice<string> fetches,
|
||||
gtl::ArraySlice<string> target_nodes,
|
||||
std::unordered_map<string, Graph*>* outputs) {
|
||||
Graph graph(OpRegistry::Global());
|
||||
@ -462,8 +462,7 @@ Status DirectSession::CreateGraphs(gtl::ArraySlice<string> feeds,
|
||||
// allow.
|
||||
device_opts.allow_internal_ops = true;
|
||||
device_opts.expect_device_spec = true;
|
||||
Status s =
|
||||
ConvertGraphDefToGraph(device_opts, graph_def, device_graph);
|
||||
Status s = ConvertGraphDefToGraph(device_opts, graph_def, device_graph);
|
||||
if (!s.ok()) {
|
||||
delete device_graph;
|
||||
// Also delete other graphs created during the loop.
|
||||
|
||||
@ -6,17 +6,17 @@
|
||||
#include <unordered_map>
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/core/platform/thread_annotations.h"
|
||||
#include "tensorflow/core/common_runtime/device_mgr.h"
|
||||
#include "tensorflow/core/common_runtime/device_set.h"
|
||||
#include "tensorflow/core/common_runtime/executor.h"
|
||||
#include "tensorflow/core/framework/cancellation.h"
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/public/session.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/platform/thread_annotations.h"
|
||||
#include "tensorflow/core/public/session.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -5,6 +5,7 @@
|
||||
#include <unordered_map>
|
||||
#include <vector>
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/framework/tensor_testutil.h"
|
||||
@ -18,7 +19,6 @@
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/util/device_name_utils.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_EIGEN_THREAD_POOL_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_EIGEN_THREAD_POOL_H_
|
||||
|
||||
#include "tensorflow/core/lib/core/threadpool.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/lib/core/threadpool.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,12 +1,12 @@
|
||||
#include "tensorflow/core/common_runtime/executor.h"
|
||||
|
||||
#include <atomic>
|
||||
#include <deque>
|
||||
#include <memory>
|
||||
#include <string>
|
||||
#include <unordered_map>
|
||||
#include <unordered_set>
|
||||
#include <vector>
|
||||
#include <deque>
|
||||
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include "tensorflow/core/framework/cancellation.h"
|
||||
|
||||
@ -4,8 +4,8 @@
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/framework/rendezvous.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/lib/core/notification.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_allocator_retry.h"
|
||||
#include "tensorflow/core/public/env.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/public/env.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,11 +1,11 @@
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_allocator_retry.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/lib/core/notification.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/platform/thread_annotations.h"
|
||||
#include "tensorflow/core/public/env.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_debug_allocator.h"
|
||||
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
#include "tensorflow/stream_executor/multi_platform_manager.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
|
||||
namespace gpu = ::perftools::gputools;
|
||||
|
||||
|
||||
@ -6,9 +6,9 @@
|
||||
#include <unordered_map>
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/visitable_allocator.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/visitable_allocator.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -5,14 +5,14 @@
|
||||
#include <algorithm>
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/lib/gtl/inlined_vector.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h"
|
||||
#include "tensorflow/stream_executor/multi_platform_manager.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
#include "tensorflow/core/lib/gtl/inlined_vector.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace gpu = ::perftools::gputools;
|
||||
|
||||
|
||||
@ -5,7 +5,10 @@
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_DEVICE_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_DEVICE_H_
|
||||
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/common_runtime/device_factory.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h"
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#include "tensorflow/core/common_runtime/local_device.h"
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
@ -15,9 +18,6 @@
|
||||
#include "tensorflow/core/public/session_options.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h"
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -2,10 +2,10 @@
|
||||
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h"
|
||||
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
#include "tensorflow/stream_executor/multi_platform_manager.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
|
||||
namespace gpu = ::perftools::gputools;
|
||||
|
||||
|
||||
@ -2,13 +2,13 @@
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/stream_executor/multi_platform_manager.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/strings/numbers.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace gpu = ::perftools::gputools;
|
||||
|
||||
|
||||
@ -5,12 +5,12 @@
|
||||
#include <algorithm>
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/lib/gtl/inlined_vector.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
#include "tensorflow/core/lib/gtl/inlined_vector.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace gpu = ::perftools::gputools;
|
||||
|
||||
|
||||
@ -1,7 +1,12 @@
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_util.h"
|
||||
|
||||
//#include "base/commandlineflags.h"
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/dma_helper.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/process_state.h"
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
@ -15,12 +20,7 @@
|
||||
#include "tensorflow/core/platform/tensor_coding.h"
|
||||
#include "tensorflow/core/platform/tracing.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/dma_helper.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/process_state.h"
|
||||
#include "tensorflow/core/util/util.h"
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
|
||||
#include "tensorflow/core/platform/stream_executor_util.h"
|
||||
|
||||
|
||||
@ -1,11 +1,11 @@
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_UTIL_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_UTIL_H_
|
||||
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/dma_helper.h"
|
||||
#include "tensorflow/stream_executor/device_memory.h"
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/dma_helper.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
|
||||
|
||||
@ -1,9 +1,9 @@
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_util.h"
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_util.h"
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -6,10 +6,10 @@
|
||||
|
||||
#include <map>
|
||||
|
||||
//#include "prodkernel/api/base/numa.h"
|
||||
#include "tensorflow/core/lib/strings/numbers.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
//#include "prodkernel/api/base/numa.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -8,11 +8,11 @@
|
||||
#include <atomic>
|
||||
#include <map>
|
||||
#include <memory>
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/visitable_allocator.h"
|
||||
#include "tensorflow/core/lib/core/bits.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/visitable_allocator.h"
|
||||
#include "tensorflow/stream_executor/stream_executor.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,15 +1,15 @@
|
||||
#include "tensorflow/core/common_runtime/gpu/process_state.h"
|
||||
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
#include "tensorflow/stream_executor/multi_platform_manager.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_debug_allocator.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_init.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/gpu_region_allocator.h"
|
||||
#include "tensorflow/core/common_runtime/gpu/pool_allocator.h"
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/stream_executor/multi_platform_manager.h"
|
||||
|
||||
#if defined(PLATFORM_GOOGLE)
|
||||
DEFINE_bool(record_mem_types, false,
|
||||
|
||||
@ -4,17 +4,17 @@
|
||||
#include "tensorflow/core/common_runtime/device_factory.h"
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/framework/op_segment.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/lib/core/notification.h"
|
||||
#include "tensorflow/core/lib/core/threadpool.h"
|
||||
#include "tensorflow/core/lib/strings/str_util.h"
|
||||
#include "tensorflow/core/util/device_name_utils.h"
|
||||
#include "tensorflow/core/lib/core/notification.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/platform/test_benchmark.h"
|
||||
#include "tensorflow/core/public/session_options.h"
|
||||
#include "tensorflow/core/util/device_name_utils.h"
|
||||
|
||||
#if defined(PLATFORM_GOOGLE)
|
||||
DECLARE_bool(brain_gpu_use_bfc_allocator);
|
||||
@ -71,9 +71,7 @@ Benchmark::Benchmark(const string& device, Graph* g,
|
||||
|
||||
TF_CHECK_OK(NewLocalExecutor(
|
||||
{
|
||||
device_,
|
||||
nullptr,
|
||||
false,
|
||||
device_, nullptr, false,
|
||||
[this](const NodeDef& ndef, OpKernel** kernel) {
|
||||
return CreateNonCachedKernel(device_, nullptr, ndef, kernel);
|
||||
},
|
||||
|
||||
@ -4,10 +4,10 @@
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/lib/core/threadpool.h"
|
||||
#include "tensorflow/core/common_runtime/executor.h"
|
||||
#include "tensorflow/core/graph/testlib.h"
|
||||
#include "tensorflow/core/lib/core/threadpool.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
@ -1,12 +1,12 @@
|
||||
#define EIGEN_USE_THREADS
|
||||
|
||||
#include "tensorflow/core/common_runtime/eigen_thread_pool.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/common_runtime/local_device.h"
|
||||
#include "tensorflow/core/lib/core/threadpool.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/public/session_options.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
#include "tensorflow/core/public/session.h"
|
||||
|
||||
#include "tensorflow/core/public/session_options.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/public/session_options.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -7,9 +7,9 @@
|
||||
#include "tensorflow/core/common_runtime/device_set.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/public/session_options.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/util/device_name_utils.h"
|
||||
#include "tensorflow/core/public/session_options.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -5,6 +5,7 @@
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/common_runtime/device_set.h"
|
||||
#include "tensorflow/core/framework/device_attributes.pb.h"
|
||||
@ -20,7 +21,6 @@
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -6,8 +6,8 @@
|
||||
|
||||
#include <limits>
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include <algorithm>
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
namespace tensorflow {
|
||||
|
||||
TEST(CPUAllocatorTest, Simple) {
|
||||
|
||||
@ -4,11 +4,11 @@
|
||||
#include <string>
|
||||
#include "tensorflow/core/framework/attr_value.pb.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#ifndef TENSORFLOW_FRAMEWORK_BFLOAT16_H_
|
||||
#define TENSORFLOW_FRAMEWORK_BFLOAT16_H_
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
// Compact 16-bit encoding of floating point numbers. This representation uses
|
||||
// 1 bit for the sign, 8 bits for the exponent and 7 bits for the mantissa. It
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
#include "tensorflow/core/framework/bfloat16.h"
|
||||
|
||||
#include "tensorflow/core/platform/test_benchmark.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/platform/test_benchmark.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#include "tensorflow/core/framework/cancellation.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/lib/core/notification.h"
|
||||
#include "tensorflow/core/lib/core/threadpool.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,9 +1,9 @@
|
||||
#ifndef TENSORFLOW_FRAMEWORK_CONTROL_FLOW_H_
|
||||
#define TENSORFLOW_FRAMEWORK_CONTROL_FLOW_H_
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/lib/hash/hash.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -6,11 +6,11 @@
|
||||
|
||||
#include "tensorflow/core/framework/device_attributes.pb.h"
|
||||
#include "tensorflow/core/framework/tensor.pb.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/core/refcount.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
namespace Eigen {
|
||||
class ThreadPoolDevice;
|
||||
|
||||
@ -84,8 +84,7 @@ Status FakeInputImpl::AddInputToBuilder() {
|
||||
} else {
|
||||
if (!dt_specified_ && !arg_->type_list_attr().empty()) {
|
||||
DataTypeVector dts;
|
||||
Status status =
|
||||
GetNodeAttr(*node_def_, arg_->type_list_attr(), &dts);
|
||||
Status status = GetNodeAttr(*node_def_, arg_->type_list_attr(), &dts);
|
||||
if (!status.ok()) {
|
||||
return errors::InvalidArgument(
|
||||
"Could not infer list of types for input '", arg_->name(), "': ",
|
||||
|
||||
@ -1,4 +1,5 @@
|
||||
#include "tensorflow/core/framework/function.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/function.pb.h"
|
||||
#include "tensorflow/core/framework/function_testlib.h"
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
@ -10,7 +11,6 @@
|
||||
#include "tensorflow/core/lib/strings/str_util.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#include "tensorflow/core/framework/kernel_def_builder.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/kernel_def.pb.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -49,8 +49,7 @@ bool NodeDefBuilder::NextArgAvailable() {
|
||||
|
||||
NodeDefBuilder& NodeDefBuilder::Input(FakeInputFunctor fake_input) {
|
||||
if (NextArgAvailable()) {
|
||||
Status status =
|
||||
fake_input(*op_def_, inputs_specified_, node_def_, this);
|
||||
Status status = fake_input(*op_def_, inputs_specified_, node_def_, this);
|
||||
if (!status.ok()) errors_.push_back(status.error_message());
|
||||
}
|
||||
return *this;
|
||||
|
||||
@ -8,9 +8,9 @@
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/framework/op_def.pb.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,13 +1,13 @@
|
||||
#include "tensorflow/core/framework/node_def_builder.h"
|
||||
|
||||
#include <memory>
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/fake_input.h"
|
||||
#include "tensorflow/core/framework/node_def_util.h"
|
||||
#include "tensorflow/core/framework/op_def_builder.h"
|
||||
#include "tensorflow/core/framework/op_def_util.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -1,13 +1,13 @@
|
||||
#include "tensorflow/core/framework/node_def_util.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/fake_input.h"
|
||||
#include "tensorflow/core/framework/node_def_builder.h"
|
||||
#include "tensorflow/core/framework/op_def_builder.h"
|
||||
#include "tensorflow/core/framework/op_def_util.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -2,11 +2,11 @@
|
||||
#define TENSORFLOW_FRAMEWORK_NUMERIC_OP_H_
|
||||
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/types.pb.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/framework/types.pb.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,12 +1,12 @@
|
||||
#include "tensorflow/core/framework/op_def_builder.h"
|
||||
|
||||
#include "tensorflow/core/framework/op_def.pb.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/lib/strings/str_util.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/op_def.pb.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "tensorflow/core/lib/strings/str_util.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -1,12 +1,12 @@
|
||||
#include "tensorflow/core/framework/op_def_util.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/op_def.pb.h"
|
||||
#include "tensorflow/core/framework/op_def_builder.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include "tensorflow/core/lib/strings/str_util.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
@ -19,9 +19,7 @@ OpDef FromText(const string& text) {
|
||||
|
||||
class ValidateOpDefTest : public ::testing::Test {
|
||||
protected:
|
||||
Status TestProto(const string& text) {
|
||||
return ValidateOpDef(FromText(text));
|
||||
}
|
||||
Status TestProto(const string& text) { return ValidateOpDef(FromText(text)); }
|
||||
|
||||
Status TestBuilder(const OpDefBuilder& builder) {
|
||||
OpDef op_def;
|
||||
|
||||
@ -2,6 +2,7 @@
|
||||
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include "tensorflow/core/framework/attr_value_util.h"
|
||||
#include "tensorflow/core/framework/fake_input.h"
|
||||
@ -14,7 +15,6 @@
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
class DummyKernel : public tensorflow::OpKernel {
|
||||
public:
|
||||
|
||||
@ -1,15 +1,15 @@
|
||||
#include "tensorflow/core/framework/op_segment.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/framework/node_def_builder.h"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -3,12 +3,12 @@
|
||||
#include <unordered_map>
|
||||
#include <utility>
|
||||
|
||||
#include "tensorflow/core/platform/thread_annotations.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/core/notification.h"
|
||||
#include "tensorflow/core/lib/strings/str_util.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/platform/thread_annotations.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -3,11 +3,11 @@
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/lib/core/refcount.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/framework/control_flow.h"
|
||||
#include "tensorflow/core/framework/device_base.h"
|
||||
#include "tensorflow/core/lib/core/refcount.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/util/device_name_utils.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#include "tensorflow/core/framework/tensor_slice.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/lib/strings/str_util.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -2,13 +2,13 @@
|
||||
#define TENSORFLOW_FRAMEWORK_TENSOR_SLICE_H_
|
||||
|
||||
#include <string>
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/tensor_slice.pb.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "tensorflow/core/lib/gtl/inlined_vector.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,9 +1,9 @@
|
||||
#include "tensorflow/core/framework/tensor_slice.h"
|
||||
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -1,11 +1,11 @@
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/tensor_testutil.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/test_benchmark.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,10 +1,10 @@
|
||||
#ifndef TENSORFLOW_FRAMEWORK_TENSOR_TESTUTIL_H_
|
||||
#define TENSORFLOW_FRAMEWORK_TENSOR_TESTUTIL_H_
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
namespace test {
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#include "tensorflow/core/framework/tensor_util.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -2,9 +2,9 @@
|
||||
|
||||
#include <unordered_map>
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -5,6 +5,7 @@
|
||||
#include <set>
|
||||
#include <string>
|
||||
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint"
|
||||
#include "tensorflow/core/framework/bfloat16.h"
|
||||
#include "tensorflow/core/framework/numeric_types.h"
|
||||
#include "tensorflow/core/framework/types.pb.h"
|
||||
@ -13,7 +14,6 @@
|
||||
#include "tensorflow/core/lib/gtl/inlined_vector.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -3,14 +3,14 @@
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/graph/graph_def_builder.h"
|
||||
#include "tensorflow/core/graph/subgraph.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
|
||||
// TODO(josh11b): Test setting the "device" field of a NodeDef.
|
||||
// TODO(josh11b): Test that feeding won't prune targets.
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#include "tensorflow/core/graph/costutil.h"
|
||||
|
||||
#include "tensorflow/core/graph/algorithm.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/graph/costmodel.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -6,8 +6,8 @@
|
||||
|
||||
#include "tensorflow/core/graph/colors.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/regexp.h"
|
||||
#include "tensorflow/core/util/util.h"
|
||||
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
#include "tensorflow/core/graph/edgeset.h"
|
||||
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
|
||||
namespace tensorflow {
|
||||
class EdgeSetTest : public ::testing::Test {
|
||||
|
||||
@ -1,10 +1,10 @@
|
||||
#include "tensorflow/core/graph/equal_graph_def.h"
|
||||
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/node_def_util.h"
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/graph/graph_def_builder.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -1,5 +1,6 @@
|
||||
#include "tensorflow/core/graph/graph_constructor.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
@ -8,7 +9,6 @@
|
||||
#include "tensorflow/core/platform/protobuf.h"
|
||||
#include "tensorflow/core/platform/regexp.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
// TODO(josh11b): Test InitCostModel().
|
||||
// TODO(josh11b): Test setting the "device" field of a NodeDef.
|
||||
|
||||
@ -6,9 +6,9 @@
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/graph/node_builder.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -6,8 +6,8 @@
|
||||
#include <unordered_map>
|
||||
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/graph/costmodel.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -6,8 +6,8 @@
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/framework/op_def.pb.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,10 +1,10 @@
|
||||
#include "tensorflow/core/graph/node_builder.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -15,8 +15,8 @@
|
||||
#include "tensorflow/core/graph/graph_constructor.h"
|
||||
#include "tensorflow/core/graph/node_builder.h"
|
||||
#include "tensorflow/core/graph/tensor_id.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
@ -3,6 +3,7 @@
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/graph/graph_constructor.h"
|
||||
@ -15,7 +16,6 @@
|
||||
#include "tensorflow/core/platform/regexp.h"
|
||||
#include "tensorflow/core/platform/test_benchmark.h"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
// TODO(josh11b): Test setting the "device" field of a NodeDef.
|
||||
// TODO(josh11b): Test that feeding won't prune targets.
|
||||
@ -100,8 +100,8 @@ class SubgraphTest : public ::testing::Test {
|
||||
std::vector<string> targets =
|
||||
str_util::Split(targets_str, ',', str_util::SkipEmpty());
|
||||
|
||||
Status s = subgraph::RewriteGraphForExecution(subgraph, fed, fetch,
|
||||
targets, device_info_);
|
||||
Status s = subgraph::RewriteGraphForExecution(subgraph, fed, fetch, targets,
|
||||
device_info_);
|
||||
if (!s.ok()) {
|
||||
delete subgraph;
|
||||
return s.ToString();
|
||||
|
||||
@ -3,8 +3,8 @@
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -6,9 +6,9 @@
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/graph/graph.h"
|
||||
#include "tensorflow/core/graph/types.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#ifndef TENSORFLOW_GRAPH_TYPES_H_
|
||||
#define TENSORFLOW_GRAPH_TYPES_H_
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/lib/gtl/int_type.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -2,6 +2,7 @@
|
||||
#define EIGEN_USE_THREADS
|
||||
|
||||
#include <memory>
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/register_types.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
@ -10,7 +11,6 @@
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
@ -20,8 +20,8 @@ typedef Eigen::GpuDevice GPUDevice;
|
||||
template <typename Device, typename T>
|
||||
class AdjustContrastOp : public OpKernel {
|
||||
public:
|
||||
explicit AdjustContrastOp(OpKernelConstruction* context) : OpKernel(context) {
|
||||
}
|
||||
explicit AdjustContrastOp(OpKernelConstruction* context)
|
||||
: OpKernel(context) {}
|
||||
|
||||
void Compute(OpKernelContext* context) override {
|
||||
const Tensor& input = context->input(0);
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
#ifndef TENSORFLOW_KERNELS_ADJUST_CONTRAST_OP_H_
|
||||
#define TENSORFLOW_KERNELS_ADJUST_CONTRAST_OP_H_
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace functor {
|
||||
|
||||
@ -3,8 +3,8 @@
|
||||
|
||||
// Functor definitions for Aggregate ops, must be compilable by nvcc.
|
||||
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace functor {
|
||||
|
||||
@ -4,8 +4,8 @@
|
||||
|
||||
#include "tensorflow/core/kernels/aggregate_ops.h"
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -9,14 +9,14 @@
|
||||
#include "tensorflow/core/kernels/argmax_op.h"
|
||||
|
||||
#include <memory>
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/register_types.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -2,9 +2,9 @@
|
||||
#define TENSORFLOW_KERNELS_ARGMAX_OP_H_
|
||||
// Generator definition for ArgMaxOp, must be compilable by nvcc.
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -3,9 +3,9 @@
|
||||
|
||||
#define EIGEN_USE_THREADS
|
||||
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -3,13 +3,13 @@
|
||||
#define EIGEN_USE_THREADS
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks"
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -4,19 +4,19 @@
|
||||
|
||||
#include "tensorflow/core/kernels/avgpooling_op.h"
|
||||
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/numeric_op.h"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "tensorflow/core/framework/tensor_slice.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/kernels/pooling_ops_common.h"
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/util/padding.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks"
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "tensorflow/core/util/padding.h"
|
||||
|
||||
#if GOOGLE_CUDA
|
||||
#include "tensorflow/core/kernels/maxpooling_op_gpu.h"
|
||||
@ -34,13 +34,11 @@ class AvgPoolingOp : public UnaryOp<T> {
|
||||
explicit AvgPoolingOp(OpKernelConstruction* context) : UnaryOp<T>(context) {
|
||||
OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
|
||||
OP_REQUIRES(context, ksize_.size() == 4,
|
||||
errors::InvalidArgument(
|
||||
"Sliding window ksize field must "
|
||||
errors::InvalidArgument("Sliding window ksize field must "
|
||||
"specify 4 dimensions"));
|
||||
OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_));
|
||||
OP_REQUIRES(context, stride_.size() == 4,
|
||||
errors::InvalidArgument(
|
||||
"Sliding window stride field must "
|
||||
errors::InvalidArgument("Sliding window stride field must "
|
||||
"specify 4 dimensions"));
|
||||
OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_));
|
||||
OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
|
||||
@ -56,8 +54,7 @@ class AvgPoolingOp : public UnaryOp<T> {
|
||||
return;
|
||||
}
|
||||
OP_REQUIRES(context, params.depth_window == 1,
|
||||
errors::Unimplemented(
|
||||
"Non-spatial pooling is not "
|
||||
errors::Unimplemented("Non-spatial pooling is not "
|
||||
"yet supported. Volunteers? :)"));
|
||||
|
||||
// For avgpooling, tensor_in should have 4 dimensions.
|
||||
@ -123,13 +120,11 @@ class AvgPoolingGradOp : public OpKernel {
|
||||
explicit AvgPoolingGradOp(OpKernelConstruction* context) : OpKernel(context) {
|
||||
OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
|
||||
OP_REQUIRES(context, ksize_.size() == 4,
|
||||
errors::InvalidArgument(
|
||||
"Sliding window ksize field must "
|
||||
errors::InvalidArgument("Sliding window ksize field must "
|
||||
"specify 4 dimensions"));
|
||||
OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_));
|
||||
OP_REQUIRES(context, stride_.size() == 4,
|
||||
errors::InvalidArgument(
|
||||
"Sliding window strides field must "
|
||||
errors::InvalidArgument("Sliding window strides field must "
|
||||
"specify 4 dimensions"));
|
||||
OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_));
|
||||
OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
|
||||
@ -141,10 +136,10 @@ class AvgPoolingGradOp : public OpKernel {
|
||||
const Tensor& tensor_in_shape = context->input(0);
|
||||
const Tensor& out_backprop = context->input(1);
|
||||
// For avgpooling, tensor_in_shape should have 1 dimension, and 4 elements.
|
||||
OP_REQUIRES(context, tensor_in_shape.dims() == 1 &&
|
||||
tensor_in_shape.NumElements() == 4,
|
||||
errors::InvalidArgument(
|
||||
"out_backprop must be 1-dimensional and 4 "
|
||||
OP_REQUIRES(
|
||||
context,
|
||||
tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4,
|
||||
errors::InvalidArgument("out_backprop must be 1-dimensional and 4 "
|
||||
"elements"));
|
||||
// For avgpooling, out_backprop should have 4 dimensions.
|
||||
OP_REQUIRES(context, out_backprop.dims() == 4,
|
||||
@ -178,8 +173,7 @@ class AvgPoolingGradOp : public OpKernel {
|
||||
//
|
||||
// Spatial pooling is when depth_window = 1
|
||||
OP_REQUIRES(context, depth_window == 1,
|
||||
errors::Unimplemented(
|
||||
"Non-spatial pooling is not "
|
||||
errors::Unimplemented("Non-spatial pooling is not "
|
||||
"yet supported. Volunteers? :)"));
|
||||
|
||||
int out_height, out_width, pad_rows, pad_cols;
|
||||
|
||||
@ -2,9 +2,9 @@
|
||||
#define TENSORFLOW_KERNELS_AVGPOOLING_OP_H_
|
||||
// Functor definition for AvgPoolingOp, must be compilable by nvcc.
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace functor {
|
||||
|
||||
@ -14,8 +14,8 @@
|
||||
#include "tensorflow/core/util/work_sharder.h"
|
||||
|
||||
#if GOOGLE_CUDA
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#endif // GOOGLE_CUDA
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
@ -3,11 +3,11 @@
|
||||
#define EIGEN_USE_THREADS
|
||||
|
||||
#include "tensorflow/core/framework/numeric_op.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/register_types.h"
|
||||
#include "tensorflow/core/kernels/batch_norm_op.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,8 +1,8 @@
|
||||
#ifndef TENSORFLOW_KERNELS_BATCH_NORM_OP_H_
|
||||
#define TENSORFLOW_KERNELS_BATCH_NORM_OP_H_
|
||||
// Functor definition for BatchNormOp, must be compilable by nvcc.
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace functor {
|
||||
|
||||
@ -3,11 +3,11 @@
|
||||
#define EIGEN_USE_THREADS
|
||||
|
||||
#include "tensorflow/core/framework/numeric_op.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/register_types.h"
|
||||
#include "tensorflow/core/kernels/bias_op.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -2,8 +2,8 @@
|
||||
#define TENSORFLOW_KERNELS_BIAS_OP_H_
|
||||
// Functor definition for BiasOp, must be compilable by nvcc.
|
||||
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace functor {
|
||||
|
||||
@ -1,10 +1,10 @@
|
||||
#ifndef TENSORFLOW_KERNELS_CAST_OP_H_
|
||||
#define TENSORFLOW_KERNELS_CAST_OP_H_
|
||||
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/bfloat16.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace functor {
|
||||
|
||||
@ -1,12 +1,12 @@
|
||||
#include "tensorflow/core/framework/fake_input.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/common_runtime/kernel_benchmark_testlib.h"
|
||||
#include "tensorflow/core/framework/node_def_builder.h"
|
||||
#include "tensorflow/core/framework/types.pb.h"
|
||||
#include "tensorflow/core/common_runtime/kernel_benchmark_testlib.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/kernels/ops_testutil.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/platform/test_benchmark.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -8,8 +8,8 @@
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
#if GOOGLE_CUDA
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#endif // GOOGLE_CUDA
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,14 +1,14 @@
|
||||
#if GOOGLE_CUDA
|
||||
#define EIGEN_USE_GPU
|
||||
|
||||
#include <stdio.h>
|
||||
#include <assert.h>
|
||||
#include <stdio.h>
|
||||
|
||||
#include <math.h>
|
||||
#include <algorithm>
|
||||
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -4,6 +4,7 @@
|
||||
|
||||
#include <cmath>
|
||||
|
||||
#include "third_party/eigen3/Eigen/Cholesky"
|
||||
#include "tensorflow/core/framework/kernel_def_builder.h"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/kernels/linalg_ops_common.h"
|
||||
@ -11,7 +12,6 @@
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "third_party/eigen3/Eigen/Cholesky"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -2,15 +2,15 @@
|
||||
|
||||
#include <vector>
|
||||
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/register_types.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/kernels/concat_op.h"
|
||||
#include "tensorflow/core/platform/port.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/public/status.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -2,19 +2,19 @@
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/common_runtime/kernel_benchmark_testlib.h"
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/framework/types.pb.h"
|
||||
#include "tensorflow/core/graph/testlib.h"
|
||||
#include "tensorflow/core/graph/node_builder.h"
|
||||
#include "tensorflow/core/graph/testlib.h"
|
||||
#include "tensorflow/core/kernels/ops_testutil.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
#include "tensorflow/core/platform/test_benchmark.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/lib/core/status_test_util.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -4,13 +4,13 @@
|
||||
|
||||
#include "tensorflow/core/kernels/constant_op.h"
|
||||
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/register_types.h"
|
||||
#include "tensorflow/core/framework/tensor.pb.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/kernels/fill_functor.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,9 +1,9 @@
|
||||
#ifndef TENSORFLOW_KERNELS_CONSTANT_OP_H_
|
||||
#define TENSORFLOW_KERNELS_CONSTANT_OP_H_
|
||||
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
||||
@ -1,11 +1,11 @@
|
||||
#include "tensorflow/core/framework/fake_input.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/framework/node_def_builder.h"
|
||||
#include "tensorflow/core/framework/tensor_testutil.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/kernels/ops_testutil.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace tensorflow {
|
||||
namespace {
|
||||
|
||||
@ -1,9 +1,9 @@
|
||||
#ifndef TENSORFLOW_KERNELS_CONV_2D_H_
|
||||
#define TENSORFLOW_KERNELS_CONV_2D_H_
|
||||
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks"
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/tensor_types.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace functor {
|
||||
|
||||
@ -5,20 +5,20 @@
|
||||
|
||||
#include "tensorflow/core/framework/numeric_op.h"
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "tensorflow/core/framework/tensor_slice.h"
|
||||
#include "tensorflow/core/kernels/conv_2d.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/util/use_cudnn.h"
|
||||
#include "tensorflow/core/util/padding.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/public/tensor.h"
|
||||
#include "tensorflow/core/public/tensor_shape.h"
|
||||
#include "tensorflow/core/util/padding.h"
|
||||
#include "tensorflow/core/util/use_cudnn.h"
|
||||
|
||||
#if GOOGLE_CUDA
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#include "tensorflow/stream_executor/stream.h"
|
||||
#include "tensorflow/core/common_runtime/gpu_device_context.h"
|
||||
#endif // GOOGLE_CUDA
|
||||
|
||||
namespace tensorflow {
|
||||
@ -271,16 +271,15 @@ class Conv2DFastBackpropInputOp : public OpKernel {
|
||||
: OpKernel(context) {
|
||||
OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
|
||||
OP_REQUIRES(context, strides_.size() == 4,
|
||||
errors::InvalidArgument(
|
||||
"Sliding window strides field must "
|
||||
errors::InvalidArgument("Sliding window strides field must "
|
||||
"specify 4 dimensions"));
|
||||
OP_REQUIRES(context, strides_[1] == strides_[2],
|
||||
errors::InvalidArgument(
|
||||
"Current implementation only supports equal length "
|
||||
"strides in the row and column dimensions."));
|
||||
OP_REQUIRES(context, (strides_[0] == 1 && strides_[3] == 1),
|
||||
errors::InvalidArgument(
|
||||
"Current implementation does not yet support "
|
||||
OP_REQUIRES(
|
||||
context, (strides_[0] == 1 && strides_[3] == 1),
|
||||
errors::InvalidArgument("Current implementation does not yet support "
|
||||
"strides in the batch and depth dimensions."));
|
||||
OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_));
|
||||
}
|
||||
@ -444,16 +443,15 @@ class Conv2DFastBackpropFilterOp : public OpKernel {
|
||||
: OpKernel(context) {
|
||||
OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
|
||||
OP_REQUIRES(context, strides_.size() == 4,
|
||||
errors::InvalidArgument(
|
||||
"Sliding window strides field must "
|
||||
errors::InvalidArgument("Sliding window strides field must "
|
||||
"specify 4 dimensions"));
|
||||
OP_REQUIRES(context, strides_[1] == strides_[2],
|
||||
errors::InvalidArgument(
|
||||
"Current implementation only supports equal length "
|
||||
"strides in the row and column dimensions."));
|
||||
OP_REQUIRES(context, (strides_[0] == 1 && strides_[3] == 1),
|
||||
errors::InvalidArgument(
|
||||
"Current implementation does not yet support "
|
||||
OP_REQUIRES(
|
||||
context, (strides_[0] == 1 && strides_[3] == 1),
|
||||
errors::InvalidArgument("Current implementation does not yet support "
|
||||
"strides in the batch and depth dimensions."));
|
||||
OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_));
|
||||
}
|
||||
@ -668,16 +666,15 @@ class Conv2DSlowBackpropInputOp : public OpKernel {
|
||||
: OpKernel(context) {
|
||||
OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
|
||||
OP_REQUIRES(context, strides_.size() == 4,
|
||||
errors::InvalidArgument(
|
||||
"Sliding window strides field must "
|
||||
errors::InvalidArgument("Sliding window strides field must "
|
||||
"specify 4 dimensions"));
|
||||
OP_REQUIRES(context, strides_[1] == strides_[2],
|
||||
errors::InvalidArgument(
|
||||
"Current implementation only supports equal length "
|
||||
"strides in the row and column dimensions."));
|
||||
OP_REQUIRES(context, (strides_[0] == 1 && strides_[3] == 1),
|
||||
errors::InvalidArgument(
|
||||
"Current implementation does not yet support "
|
||||
OP_REQUIRES(
|
||||
context, (strides_[0] == 1 && strides_[3] == 1),
|
||||
errors::InvalidArgument("Current implementation does not yet support "
|
||||
"strides in the batch and depth dimensions."));
|
||||
OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_));
|
||||
use_cudnn_ &= CanUseCudnn();
|
||||
@ -901,16 +898,15 @@ class Conv2DSlowBackpropFilterOp : public OpKernel {
|
||||
: OpKernel(context) {
|
||||
OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
|
||||
OP_REQUIRES(context, strides_.size() == 4,
|
||||
errors::InvalidArgument(
|
||||
"Sliding window strides field must "
|
||||
errors::InvalidArgument("Sliding window strides field must "
|
||||
"specify 4 dimensions"));
|
||||
OP_REQUIRES(context, strides_[1] == strides_[2],
|
||||
errors::InvalidArgument(
|
||||
"Current implementation only supports equal length "
|
||||
"strides in the row and column dimensions."));
|
||||
OP_REQUIRES(context, (strides_[0] == 1 && strides_[3] == 1),
|
||||
errors::InvalidArgument(
|
||||
"Current implementation does not yet support "
|
||||
OP_REQUIRES(
|
||||
context, (strides_[0] == 1 && strides_[3] == 1),
|
||||
errors::InvalidArgument("Current implementation does not yet support "
|
||||
"strides in the batch and depth dimensions."));
|
||||
OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_));
|
||||
use_cudnn_ &= CanUseCudnn();
|
||||
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user