Searched refs:add_md (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 164 auto add_md = in Compute() local
178 add_md = dst_md = in Compute()
183 auto fuse_add_src_ = memory(add_md, this->cpu_engine_, add_buf); in Compute()
186 ReorderPd(this->cpu_engine_, add_md, this->cpu_engine_, dst_md); in Compute()
|
| D | mkl_conv_ops.cc | 902 auto add_md = in AllocateOutputTensor() local
914 add_md = dst_md = in AllocateOutputTensor()
918 fuse_add_src_.reset(new memory(add_md, this->cpu_engine_, add_buf)); in AllocateOutputTensor()
921 ReorderPd(this->cpu_engine_, add_md, this->cpu_engine_, dst_md); in AllocateOutputTensor()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 26 #define ADD_MD add_md
|