| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | split_module.py | 47 split_callback: Callable[[Node], int],
236 partition_name = str(split_callback(node))
300 grad_regions[active_grad] = set({split_callback(node)})
305 autocast_regions[node] = set({split_callback(node)})
309 autocast_regions[node.args[0]].add(split_callback(node))
314 grad_regions[active_grad].add(split_callback(node))
317 autocast_regions[a].add(split_callback(node))
346 pid = split_callback(node)
|
| /aosp_15_r20/external/cronet/net/disk_cache/simple/ |
| H A D | simple_backend_impl.cc | 135 auto split_callback = base::SplitOnceCallback(std::move(operation_callback)); in RunOperationAndCallback() local
137 std::move(operation).Run(std::move(split_callback.first)); in RunOperationAndCallback()
138 if (operation_result != net::ERR_IO_PENDING && split_callback.second) in RunOperationAndCallback()
139 std::move(split_callback.second).Run(operation_result); in RunOperationAndCallback()
150 auto split_callback = base::SplitOnceCallback(std::move(operation_callback)); in RunEntryResultOperationAndCallback() local
152 std::move(operation).Run(std::move(split_callback.first)); in RunEntryResultOperationAndCallback()
154 split_callback.second) { in RunEntryResultOperationAndCallback()
155 std::move(split_callback.second).Run(std::move(operation_result)); in RunEntryResultOperationAndCallback()
605 auto split_callback = base::SplitOnceCallback(std::move(callback)); in OpenNextEntryImpl() local
606 callback = std::move(split_callback.first); in OpenNextEntryImpl()
[all …]
|
| /aosp_15_r20/external/cronet/net/test/embedded_test_server/ |
| H A D | http1_connection.cc | 157 auto split_callback = base::SplitOnceCallback(std::move(callback)); in SendInternal() local
158 callback = std::move(split_callback.first); in SendInternal()
163 std::move(split_callback.second), buf), in SendInternal()
|
| /aosp_15_r20/external/cronet/net/socket/ |
| H A D | websocket_transport_client_socket_pool.cc | 437 auto split_callback = base::SplitOnceCallback(std::move(request.callback)); in ActivateStalledRequest() local
444 RespectLimits::ENABLED, request.handle, std::move(split_callback.first), in ActivateStalledRequest()
450 InvokeUserCallbackLater(request.handle, std::move(split_callback.second), in ActivateStalledRequest()
|
| /aosp_15_r20/external/cronet/components/nacl/browser/ |
| H A D | pnacl_host.cc | 601 auto split_callback = base::SplitOnceCallback(std::move(callback)); in ClearTranslationCacheEntriesBetween() local
605 std::move(split_callback.first))); in ClearTranslationCacheEntriesBetween()
607 OnEntriesDoomed(std::move(split_callback.second), rv); in ClearTranslationCacheEntriesBetween()
|
| /aosp_15_r20/external/cronet/net/http/ |
| H A D | mock_http_cache.cc | 445 auto split_callback = base::SplitOnceCallback(std::move(callback)); in OpenOrCreateEntry() local
446 result = OpenEntry(key, request_priority, std::move(split_callback.first)); in OpenOrCreateEntry()
452 result = CreateEntry(key, request_priority, std::move(split_callback.second)); in OpenOrCreateEntry()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx_experimental.py | 804 def split_callback(n): function
815 split = split_module(traced, mod, split_callback)
833 def split_callback(n): function
840 split = split_module(traced, mod, split_callback)
1223 def split_callback(n : torch.fx.Node): function
1232 my_module_traced, my_module, split_callback, qualname_map
|
| /aosp_15_r20/external/pytorch/torch/distributed/pipelining/ |
| H A D | _IR.py | 721 def split_callback(n: fx.Node): function
733 split = split_module(traced, mod, split_callback) # type: ignore[arg-type]
|
| /aosp_15_r20/external/cronet/net/quic/ |
| H A D | quic_session_pool.cc | 769 auto split_callback = base::SplitOnceCallback(std::move(callback)); in ConnectAndConfigureSocket() local
773 std::move(split_callback.first), socket, socket_tag); in ConnectAndConfigureSocket()
788 FinishConnectAndConfigureSocket(std::move(split_callback.second), socket, in ConnectAndConfigureSocket()
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | TestFXAPIBackwardCompatibility.test_function_back_compat-fx_backcompat_function_signatures.expect | 67 …fx.graph_module.GraphModule, root_m: torch.nn.modules.module.Module, split_callback: Callable[[tor…
|