Searched defs:gpu_subgraph (Results 1 – 9 of 9) sorted by relevance
32 GPUOperationsSubgraph* gpu_subgraph) { in SelectDefault()
85 GPUOperationsSubgraph* gpu_subgraph) { in WinogradFromNode()162 int dst_id, GPUOperationsSubgraph* gpu_subgraph, in AddDynamicConv()247 GPUOperationsSubgraph* gpu_subgraph) { in AddConvSharedWeights()314 GPUOperationsSubgraph* gpu_subgraph) { in GPUOperationFromNodePart0()822 GPUOperationsSubgraph* gpu_subgraph) { in GPUOperationFromNode()
38 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in GPUSubgraphFromGraph()
47 GPUOperationsSubgraph* gpu_subgraph) { in InitSingleOpSubgraph()
547 GPUOperationsSubgraph* gpu_subgraph) { in Finalize()620 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryDepthwiseConvPlus1x1Conv()
212 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryFusedPointwiseConv()
304 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryFCFCAdd()
343 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryMeanStdDevNormalization()
300 GPUOperationsSubgraph gpu_subgraph; in ConvertOperations() local