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

[MLIR/GPU] Fix broken builds.

Browse Source 
PiperOrigin-RevId: 303806819
Change-Id: Ib9c7d9e365a5ba63ec91c771b256ed8d972e46a3
This commit is contained in:
Tim Shen 2020-03-30 13:09:47 -07:00 committed by TensorFlower Gardener
parent b3d1e0f2ea
commit 763b3b8ee4
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/BUILD
View File

@ -47,7 +47,6 @@ cc_library(
deps = [
"//tensorflow/core:lib",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/types:span",
"@llvm-project//llvm:support",
"@llvm-project//mlir:Affine",

2
tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/conv_emitter_transforms.h
View File

@ -16,11 +16,11 @@ limitations under the License.
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_MLIR_GPU_EXPERIMENTAL_CONV_EMITTER_CONV_EMITTER_TRANSFORMS_H_
#define TENSORFLOW_COMPILER_XLA_SERVICE_MLIR_GPU_EXPERIMENTAL_CONV_EMITTER_CONV_EMITTER_TRANSFORMS_H_
#include "absl/base/integral_types.h"
#include "absl/types/span.h"
#include "mlir/Dialect/Affine/IR/AffineOps.h" // from @llvm-project
#include "mlir/Dialect/StandardOps/IR/Ops.h" // from @llvm-project
#include "mlir/IR/Operation.h" // from @llvm-project
#include "tensorflow/core/platform/types.h"
namespace xla {
namespace mlir_gpu {