Searched refs:MklLayoutRewritePass (Results 1 – 1 of 1) sorted by relevance
239 class MklLayoutRewritePass : public GraphOptimizationPass { class241 MklLayoutRewritePass() { in MklLayoutRewritePass() function in tensorflow::MklLayoutRewritePass1070 std::tuple<bool, std::vector<Node*>, const MklLayoutRewritePass::FusionInfo>1075 const MklLayoutRewritePass::FusionInfo fi);1544 MklLayoutRewritePass::ConstStringsInfo MklLayoutRewritePass::csinfo_;1552 REGISTER_OPTIMIZATION(kMklLayoutRewritePassGroup, 1, MklLayoutRewritePass);1579 void MklLayoutRewritePass::GetNodesProducingTFTensorList( in GetNodesProducingTFTensorList()1601 void MklLayoutRewritePass::GetDummyMklTensorNode(std::unique_ptr<Graph>* g, in GetDummyMklTensorNode()1643 void MklLayoutRewritePass::GetNodesProducingMklTensorList( in GetNodesProducingMklTensorList()1674 void MklLayoutRewritePass::GetNodeProducingMklTensor( in GetNodeProducingMklTensor()[all …]