mirror of
https://github.com/tensorflow/tensorflow.git
synced 2026-10-26 00:09:51 +08:00
Automated Code Change
PiperOrigin-RevId: 1003612014
This commit is contained in:
1 parent
8e5806467f
commit
c28d1fde54
3 files changed
+19
-6
No files matched your search
@@ -175,7 +175,12 @@ IMAGE_TEST_DEPS = [
|
||||
tf_kernel_library(
|
||||
name = "adjust_contrast_op",
|
||||
prefix = "adjust_contrast_op",
|
||||
deps = IMAGE_DEPS + ["//tensorflow/core/util:determinism_for_kernels"],
|
||||
deps = IMAGE_DEPS + [
|
||||
"//tensorflow/core/util:determinism_for_kernels",
|
||||
"@com_google_absl//absl/log:check",
|
||||
"@com_google_absl//absl/status",
|
||||
"@com_google_absl//absl/strings",
|
||||
],
|
||||
)
|
||||
|
||||
cc_library(
|
||||
@@ -303,9 +308,10 @@ tf_kernel_library(
|
||||
deps = [
|
||||
"//tensorflow/core:framework",
|
||||
"//tensorflow/core:lib",
|
||||
"//tensorflow/core/kernels:eigen_helpers",
|
||||
"//tensorflow/core/kernels:ops_util",
|
||||
"//tensorflow/core/util:overflow",
|
||||
"@com_google_absl//absl/status",
|
||||
"@com_google_absl//absl/strings",
|
||||
"@eigen_archive//:eigen3",
|
||||
],
|
||||
)
|
||||
|
||||
@@ -14,18 +14,21 @@ limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
// See docs in ../ops/image_ops.cc
|
||||
#include <algorithm>
|
||||
#include <cstring>
|
||||
|
||||
#include "absl/log/check.h"
|
||||
#include "absl/status/status.h"
|
||||
#include "absl/strings/str_cat.h"
|
||||
#define EIGEN_USE_THREADS
|
||||
|
||||
#include "tensorflow/core/kernels/image/adjust_contrast_op.h"
|
||||
|
||||
#include <memory>
|
||||
|
||||
#include "unsupported/Eigen/CXX11/Tensor" // from @eigen_archive
|
||||
#include "tensorflow/core/framework/op_kernel.h"
|
||||
#include "tensorflow/core/framework/register_types.h"
|
||||
#include "tensorflow/core/framework/tensor.h"
|
||||
#include "tensorflow/core/framework/tensor_shape.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/kernels/image/adjust_contrast_op.h"
|
||||
#include "tensorflow/core/lib/core/status.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/util/determinism.h"
|
||||
|
||||
@@ -21,6 +21,10 @@ Rates are not supported as of now, but the comments hint how to edit the code
|
||||
when rates are to be added.
|
||||
*/
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "absl/status/status.h"
|
||||
#include "absl/strings/str_cat.h"
|
||||
#define USE_EIGEN_TENSOR
|
||||
#define EIGEN_USE_THREADS
|
||||
|
||||
|
||||
Reference in new issue
Block a user