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