Home
last modified time | relevance | path

Searched refs:tensor_key (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/cuda/
H A D_memory_viz.py454 def allocate(size, tensor_key, version, during_trace=True): argument
455 device = to_device(tensor_key.device)
456 addr = tensor_key.storage.ptr
462 category = memory_profile._categories.get(tensor_key, version)
464 stack = allocation_stacks.get(tensor_key, ())
484 for time, action, (tensor_key, version), size in memory_profile.timeline:
485 if not isinstance(tensor_key, TensorKey):
488 kv_to_elem[(tensor_key, version)] = allocate(size, tensor_key, version)
490 free(kv_to_elem.pop((tensor_key, version)), to_device(tensor_key.device))
492 free(kv_to_elem.pop((tensor_key, version)), to_device(tensor_key.device))
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Dgrpc_debug_server.py256 tensor_key = "%s_%s_%d" % (device_name, debug_node_name, timestamp)
258 if tensor_key not in tensor_chunks:
259 tensor_chunks[tensor_key] = [None] * num_chunks
261 chunks = tensor_chunks[tensor_key]
271 del tensor_chunks[tensor_key]
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpad_mm.py273 def tensor_key(t): function
286 tensor_key(mat1),
287 tensor_key(mat2),
291 input if input is None else tensor_key(input),
/aosp_15_r20/external/federated-compute/fcp/client/http/
H A Dhttp_secagg_send_to_server_impl_test.cc72 std::string tensor_key = "tensor_1"; in TEST() local
76 secure_aggregands[tensor_key] = secure_aggregand_execution_info; in TEST()
78 auto modulus = delegate.GetModulus(tensor_key); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/security/advisory/
H A Dtfsa-2022-084.md7 …lob/f3b9bf4c3c0597563b289c0512e98d4ce81f886e/tensorflow/core/framework/tensor_key.h#L53-L64) used …
/aosp_15_r20/external/tensorflow/tensorflow/python/tools/
H A Dsaved_model_cli.py447 outputs_tensor_info[tensor_key].name
448 for tensor_key in output_tensor_keys_sorted
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DTensorCompare.mm169 auto tensor_key = has_min
174 … string key = op_name + (has_min ? "_min" : "") + (has_max ? "_max" : "") + "_tensor" + tensor_key;
H A DReduceOps.mm351 …string tensor_key = cdist ? getTensorsStringKey({input_tensor, other_tensor}) : getTensorsStringKe…
352 …string key = string("norm_out_mps:") + [ns_key UTF8String] + ":" + tensor_key + ":p" + std::to_str…
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A DBUILD238 "tensor_key.h",
324 "tensor_key.h",