Searched refs:op_graph (Results 1 – 1 of 1) sorted by relevance
3868 auto op_graph = cudnn_frontend::OperationGraphBuilder() in GetCudnnFusedOperationGraph() local3872 RETURN_MSG_IF_CUDNN_ERROR(op_graph); in GetCudnnFusedOperationGraph()3892 << "\nOpGraph: " << op_graph.describe(); in GetCudnnFusedOperationGraph()3895 new cudnn_frontend::OperationGraph(std::move(op_graph))); in GetCudnnFusedOperationGraph()3902 const cudnn_frontend::OperationGraph& op_graph) { in RebuildExecutionPlan() argument3916 engine_builder.setOperationGraph(op_graph).setGlobalEngineIdx(desc.algo_id()); in RebuildExecutionPlan()4592 std::unique_ptr<cudnn_frontend::OperationGraph> op_graph, in CreateOpRunners() argument4614 .setOperationGraph(*op_graph) in CreateOpRunners()4619 .setOperationGraph(*op_graph) in CreateOpRunners()4639 .setOperationGraph(*op_graph) in CreateOpRunners()[all …]