Add types.h header to conv_ops.cc for DataTypeToEnum
PiperOrigin-RevId: 248018601
This commit is contained in:
parent
a94a09ad78
commit
37cb8bcb81
@ -36,6 +36,7 @@ limitations under the License.
|
||||
#include "tensorflow/core/framework/tensor.h"
|
||||
#include "tensorflow/core/framework/tensor_shape.h"
|
||||
#include "tensorflow/core/framework/tensor_slice.h"
|
||||
#include "tensorflow/core/framework/types.h"
|
||||
#include "tensorflow/core/kernels/conv_2d.h"
|
||||
#include "tensorflow/core/kernels/deep_conv2d.h"
|
||||
#include "tensorflow/core/kernels/ops_util.h"
|
||||
|
Loading…
Reference in New Issue
Block a user