Searched refs:fuse_add_dst_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 184 auto fuse_add_dst_ = memory(dst_md, this->cpu_engine_, dst_buf); in Compute() local
188 CreateAndExecuteReorder(reorder_desc, fuse_add_src_, fuse_add_dst_, in Compute()
|
| D | mkl_conv_ops.cc | 919 fuse_add_dst_.reset(new memory(dst_md, this->cpu_engine_, dst_buf)); in AllocateOutputTensor()
923 CreateAndExecuteReorder(reorder_desc, *fuse_add_src_, *fuse_add_dst_, in AllocateOutputTensor()
937 std::shared_ptr<mkldnn::memory> fuse_add_dst_; member in tensorflow::MklConvOp
|