Searched defs:gpu_subgraph (Results 1 – 6 of 6) sorted by relevance
35 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryDepthwiseConvPlus1x1Conv()99 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryFCFCAdd()189 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph, in GPUSubgraphFromGraph()
29 GPUOperationsSubgraph* gpu_subgraph) { in InitSingleOpSubgraph()
85 GPUOperationsSubgraph* gpu_subgraph) { in WinogradFromNode()157 GPUOperationsSubgraph* gpu_subgraph) { in GPUOperationFromNode()
32 GPUOperationsSubgraph* gpu_subgraph) { in SelectDefault()
323 GPUOperationsSubgraph gpu_subgraph; in ConvertOperations() local
231 GPUOperationsSubgraph gpu_subgraph; in Compile() local