| /external/pytorch/torch/fx/passes/ |
| 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)
|
| /external/cronet/tot/net/shared_dictionary/ |
| D | shared_dictionary_network_transaction.cc | 248 auto split_callback = base::SplitOnceCallback(base::BindOnce( in ModifyRequestHeaders() local
265 shared_dictionary_->ReadAll(std::move(split_callback.first)); in ModifyRequestHeaders()
267 std::move(split_callback.second).Run(read_result); in ModifyRequestHeaders()
288 auto split_callback = base::SplitOnceCallback(std::move(task->callback)); in OnReadSharedDictionary() local
290 Read(task->buf.get(), task->buf_len, std::move(split_callback.first)); in OnReadSharedDictionary()
292 std::move(split_callback.second).Run(ret); in OnReadSharedDictionary()
|
| /external/cronet/stable/net/shared_dictionary/ |
| D | shared_dictionary_network_transaction.cc | 248 auto split_callback = base::SplitOnceCallback(base::BindOnce( in ModifyRequestHeaders() local
265 shared_dictionary_->ReadAll(std::move(split_callback.first)); in ModifyRequestHeaders()
267 std::move(split_callback.second).Run(read_result); in ModifyRequestHeaders()
288 auto split_callback = base::SplitOnceCallback(std::move(task->callback)); in OnReadSharedDictionary() local
290 Read(task->buf.get(), task->buf_len, std::move(split_callback.first)); in OnReadSharedDictionary()
292 std::move(split_callback.second).Run(ret); in OnReadSharedDictionary()
|
| /external/cronet/tot/net/test/embedded_test_server/ |
| 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()
|
| /external/cronet/stable/net/test/embedded_test_server/ |
| 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()
|
| /external/cronet/tot/net/disk_cache/simple/ |
| 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()
601 auto split_callback = base::SplitOnceCallback(std::move(callback)); in OpenNextEntryImpl() local
602 callback = std::move(split_callback.first); in OpenNextEntryImpl()
[all …]
|
| /external/cronet/stable/net/disk_cache/simple/ |
| 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()
601 auto split_callback = base::SplitOnceCallback(std::move(callback)); in OpenNextEntryImpl() local
602 callback = std::move(split_callback.first); in OpenNextEntryImpl()
[all …]
|
| /external/cronet/tot/net/socket/ |
| 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()
|
| /external/cronet/stable/net/socket/ |
| 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()
|
| /external/cronet/tot/components/nacl/browser/ |
| D | pnacl_host.cc | 602 auto split_callback = base::SplitOnceCallback(std::move(callback)); in ClearTranslationCacheEntriesBetween() local
606 std::move(split_callback.first))); in ClearTranslationCacheEntriesBetween()
608 OnEntriesDoomed(std::move(split_callback.second), rv); in ClearTranslationCacheEntriesBetween()
|
| /external/cronet/stable/components/nacl/browser/ |
| D | pnacl_host.cc | 602 auto split_callback = base::SplitOnceCallback(std::move(callback)); in ClearTranslationCacheEntriesBetween() local
606 std::move(split_callback.first))); in ClearTranslationCacheEntriesBetween()
608 OnEntriesDoomed(std::move(split_callback.second), rv); in ClearTranslationCacheEntriesBetween()
|
| /external/cronet/tot/net/http/ |
| D | mock_http_cache.cc | 451 auto split_callback = base::SplitOnceCallback(std::move(callback)); in OpenOrCreateEntry() local
452 result = OpenEntry(key, request_priority, std::move(split_callback.first)); in OpenOrCreateEntry()
458 result = CreateEntry(key, request_priority, std::move(split_callback.second)); in OpenOrCreateEntry()
|
| /external/cronet/stable/net/http/ |
| D | mock_http_cache.cc | 451 auto split_callback = base::SplitOnceCallback(std::move(callback)); in OpenOrCreateEntry() local
452 result = OpenEntry(key, request_priority, std::move(split_callback.first)); in OpenOrCreateEntry()
458 result = CreateEntry(key, request_priority, std::move(split_callback.second)); in OpenOrCreateEntry()
|
| /external/pytorch/test/ |
| 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
|
| /external/pytorch/torch/distributed/pipelining/ |
| D | _IR.py | 721 def split_callback(n: fx.Node): function
733 split = split_module(traced, mod, split_callback) # type: ignore[arg-type]
|
| /external/cronet/tot/net/quic/ |
| D | quic_session_pool.cc | 877 auto split_callback = base::SplitOnceCallback(std::move(callback)); in ConnectAndConfigureSocket() local
881 std::move(split_callback.first), socket, socket_tag); in ConnectAndConfigureSocket()
896 FinishConnectAndConfigureSocket(std::move(split_callback.second), socket, in ConnectAndConfigureSocket()
|
| /external/cronet/stable/net/quic/ |
| D | quic_session_pool.cc | 877 auto split_callback = base::SplitOnceCallback(std::move(callback)); in ConnectAndConfigureSocket() local
881 std::move(split_callback.first), socket, socket_tag); in ConnectAndConfigureSocket()
896 FinishConnectAndConfigureSocket(std::move(split_callback.second), socket, in ConnectAndConfigureSocket()
|
| /external/pytorch/test/expect/ |
| 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…
|