|
|
This commit contains 4 main components:
1) Intel-optimized kernel implementation for Conv2D op
Implementation in kernels/mkl_conv_ops.*
2) Graph passes required to enable Conv2D optimized implementation
Implementation in graph/mkl_*. We also need a new op, MklToTf op.
Its implementation is in kernels/mkl_tfconv_op.cc.
3) Utility functions used in kernel implementation
Implementation is in common_runtime/mkl_layer_registry* and util/mkl_util.h
4) BUILD changes for Conv2D, graph passes and utility functions
|
A. Unique TensorFlower
Niranjan Hasabnis