Searched refs:MklLayoutRewritePass (Results 1 – 1 of 1) sorted by relevance
239 class MklLayoutRewritePass : public GraphOptimizationPass { class241 MklLayoutRewritePass() { in MklLayoutRewritePass() function in tensorflow::MklLayoutRewritePass1363 std::tuple<bool, std::vector<Node*>, const MklLayoutRewritePass::FusionInfo>1368 const MklLayoutRewritePass::FusionInfo fi);2060 MklLayoutRewritePass::ConstStringsInfo MklLayoutRewritePass::csinfo_;2068 REGISTER_OPTIMIZATION(kMklLayoutRewritePassGroup, 1, MklLayoutRewritePass);2089 void MklLayoutRewritePass::GetNodesProducingTFTensorList( in GetNodesProducingTFTensorList()2111 void MklLayoutRewritePass::GetDummyMklTensorNode(std::unique_ptr<Graph>* g, in GetDummyMklTensorNode()2153 void MklLayoutRewritePass::GetNodesProducingMklTensorList( in GetNodesProducingMklTensorList()2184 void MklLayoutRewritePass::GetNodeProducingMklTensor( in GetNodeProducingMklTensor()[all …]