Searched refs:reorder_desc (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 185 auto reorder_desc = in Compute() local
188 CreateAndExecuteReorder(reorder_desc, fuse_add_src_, fuse_add_dst_, in Compute()
|
| D | mkl_qmatmul_op.cc | 483 auto reorder_desc = mkldnn::reorder::primitive_desc( in GetBiasHandle() local
485 net.push_back(mkldnn::reorder(reorder_desc)); in GetBiasHandle()
|
| D | mkl_conv_ops.cc | 920 auto reorder_desc = in AllocateOutputTensor() local
923 CreateAndExecuteReorder(reorder_desc, *fuse_add_src_, *fuse_add_dst_, in AllocateOutputTensor()
1568 auto reorder_desc = in GetBiasHandle() local
1571 CreateAndExecuteReorder(reorder_desc, *input_bias_, *scaled_bias_, in GetBiasHandle()
1828 auto reorder_desc = in AllocateOutputTensor() local
1831 CreateAndExecuteReorder(reorder_desc, *summand_, *dst_, this->cpu_engine_, in AllocateOutputTensor()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1229 inline void CreateAndExecuteReorder(const ReorderPd& reorder_desc,
1234 net.push_back(mkldnn::reorder(reorder_desc));
|